zycchen 发表于 2018-8-12 06:24:42

Shell脚本里调用Python程序

  脚本背景:主管要求看门狗程序不仅仅只是看门,还要在看门成功的时候发送邮件给各个开发人员,而且必须要用公司原有的python程序作为发送邮件的主程序,所以需要在原有的看门狗程序上加一句话,而这个看门狗程序恰恰是shell程序,两种不同程序混搭交织,还有变量的混搭交织,很是让人爱恨交织。
  那个发送邮件的程序,程序名叫AutoMail.py,内容如下
from email.mime.multipart import MIMEMultipart  
from email.mime.text import MIMEText
  
from email.mime.application import MIMEApplication
  

  
mailto_list=["开发人员A@我的公司名.com","开发人员B@我的公司名.com","开发人员C@我的公司名.com"]
  
mail_host="mail.我的公司名.com"
  
mail_user="登陆发送邮箱的用户名"
  
mail_pass="这里是发送邮箱的密码"
  
mail_postfix="公司名.com"
  
mail_sender="用户名@我的公司.com"
  
def send_mail(sub, content):
  
    me=mail_sender    #为什么这里要设定一个me?写这个脚本的大哥图啥?
  
    msg = MIMEMultipart()
  
    msg['Subject'] = sub
  
    msg['From'] = me    #可以看得出,抄脚本这个大哥是从下往上抄的,me一看就是后补的,直接放mail_sender犯法?
  
    msg['To'] = ";".join(mailto_list)
  
    content1 = MIMEText(str(content), 'plain', 'utf-8')
  
    msg.attach(content1)
  
    try:
  
      s = smtplib.SMTP()
  
      s.connect(mail_host)
  
      s.login(mail_user,mail_pass)
  
      s.sendmail(me, mailto_list, msg.as_string())
  
      print('send sucess\n')
  
      s.close()
  
    except Exception as e: print(str(e))
  看门狗程序全文就不写了,这个涉及到了公司机密。
  现在的重点就是要在看门狗这个shell脚本里添加一句命令,这个命令要启动AutoMail.py这个脚本,同时还要传导出看门狗负责的模块名,这里又涉及到了在python的启动命令里添加shell变量的“难点”(其实一点也不难)。
  解决思路:Python的sys有一个属性叫sys.argv,这个属性可以拿到命令列里的参数。
  举个例子吧,假设有一个脚本叫111.py,里面的内容如下:

  当我启动这个脚本并在启动的同时传入参数的时候,输出是这样的:

  借用这个特性,就可以实现上面的要求了,我们可以把邮件的标题和正文当作两个参数随着启动AutoMail.py的语句一起执行,于是我把整个AutoMail.py的内容更改如下:
import sys  
from email.mime.multipart import MIMEMultipart
  
from email.mime.text import MIMEText
  
from email.mime.application import MIMEApplication
  

  
mailto_list=["开发人员A@我的公司名.com","开发人员B@我的公司名.com","开发人员C@我的公司名.com"]
  
mail_host="mail.我的公司名.com"
  
mail_user="登陆发送邮箱的用户名"
  
mail_pass="这里是发送邮箱的密码"
  
mail_postfix="公司名.com"
  
mail_sender="用户名@我的公司.com"
  
def send_mail(sub, content):
  
    me=mail_sender
  
    msg = MIMEMultipart()
  
    msg['Subject'] = sub
  
    msg['From'] = me
  
    msg['To'] = ";".join(mailto_list)
  
    content1 = MIMEText(str(content), 'plain', 'utf-8')
  
    msg.attach(content1)
  
    try:
  
      s = smtplib.SMTP()
  
      s.connect(mail_host)
  
      s.login(mail_user,mail_pass)
  
      s.sendmail(me, mailto_list, msg.as_string())
  
      print('send sucess\n')
  
      s.close()
  
    except Exception as e: print(str(e))
  

  
if __name__=='__main__':
  
      send_mail(*sys.argv)
  增加了一头一尾,同时在看门狗程序里添加“python /AutoMail文件的绝对路径/AutoMail.py $name已经重启 "进程重启!"”这样一句话,其中name是一个变量,就是被“看门”的模块名,保存退出之后,kill掉一个模块,让看门狗重启模块实验一下,结果很满意,如图:

  这里应该有人好奇了,明明是启动python程序,却在命令里添加了一个shell的变量,为什么python还是可以识别呢?因为虽然命令是启动python程序,但是命令本身是shell的语句啊,所以shell的语句当然可以识别shell的变量了~~~
页: [1]
查看完整版本: Shell脚本里调用Python程序