Python 申请license写入etc/rc.local

今天修正了一个问题,本来以为很简单的东西,后来思考一下,发现不完美,于是各种改,各种查资料,后来发现一种比较满意的方法,利用空闲时间记录下来

问题是这样的:机器需要申请一个license,其实就是通过check_lic命令生成一个注册码,把这个注册码通过服务器转成一个对应的序列号,最后把序列号写入/etc/rc.local

于是乎,开始干:

一开始是这样想的,第一个函数就是执行check_lic命令,返回一个注册码,第二个函数把这个注册码生成的序列号写入指定文件就ok,于是乎,代码虎虎的写完了,不超过两分钟写完:。。。。是这样的代码:

#encoding:utf-8

import subprocess
import os

def check_lic():
    p = subprocess.Popen('check_lic',shell = True,stdout=subprocess.PIPE)
    returnCode = p.wait()
    if returnCode == 0:
        out = p.stdout.readlines()
        ret = ''.join(out)
        ret = ret.strip()
        return ret
    else:
        print "check_lic run errot"
        return False
def write_in_local(lic):
    ret = '\n' + "/usr/sbin/check_lic " + lic
    with open('/etc/rc.local','a') as f:
        f.write(ret)
        f.flush()
        os.fsync(f)

if __name__ == "__main__":
    ret = check_lic()
    #这里简单测试下,直接写入ret,不进行转码
    write_in_local(ret)
在/etc/rc.local中测试了下效果,发现在文件最后面一行,写入了我们需要的东西,本以为大功告成,但是凉水来了。。。。/etc/rc.local中原来有一个check_lic开头的行,打了新的license,原来的为何不删掉?不同版本的rc.local可能最后一行会存在一个“exit 0”这样一行,那直接后面添加,就不会执行的。。。

所以,需求变了,首先,删掉原来的存在check_lic的行,第二,如果存在“exit 0”将我们要写入的东西加到"exit 0"之前行,后来我想了下,为何要留着“exit 0”,还不如删掉,反正没啥用

import subprocess
import os
import re

def check_lic():
    p = subprocess.Popen('check_lic',shell = True,stdout=subprocess.PIPE)
    returnCode = p.wait()
    if returnCode == 0:
        out = p.stdout.readlines()
        ret = ''.join(out)
        ret = ret.strip()
        #print ret
        #print type(ret)
        return ret
    else:
        digi_debug ("check_lic run error!")
        EXIT()

def write_in_local(ret):
    ret = "\n" +  "/usr/sbin/check_lic " + ret
    print ret
    new_txt = []
    reg1 = "check_lic"
    reg2 = 'exit 0'
    with open("/etc/rc.local",'r+') as title_txt:
        full_txt = title_txt.readlines()
        for line in full_txt:
            if re.search(reg1,line) or re.search(reg2,line):
                continue
            else:
                new_txt.append(line)
        title_txt.seek(0)
        title_txt.truncate(0)
        title_txt.writelines(new_txt)
        title_txt.write(ret)
        title_txt.flush()
        os.fsync(title_txt)

if __name__ == "__main__":
        ret = check_lic()
        write_in_local(ret)
搞定了。。。测试成功
前提是查资料查了一些


  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值