脚本背景:主管要求看门狗程序不仅仅只是看门,还要在看门成功的时候发送邮件给各个开发人员,而且必须要用公司原有的python程序作为发送邮件的主程序,所以需要在原有的看门狗程序上加一句话,而这个看门狗程序恰恰是shell程序,两种不同程序混搭交织,还有变量的混搭交织,很是让人爱恨交织。
那个发送邮件的程序,程序名叫AutoMail.py,内容如下
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
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的内容更改如下:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
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[
1
:])
|
增加了一头一尾,同时在看门狗程序里添加“python /AutoMail文件的绝对路径/AutoMail.py $name已经重启 "进程重启!"”这样一句话,其中name是一个变量,就是被“看门”的模块名,保存退出之后,kill掉一个模块,让看门狗重启模块实验一下,结果很满意,如图:
这里应该有人好奇了,明明是启动python程序,却在命令里添加了一个shell的变量,为什么python还是可以识别呢?因为虽然命令是启动python程序,但是命令本身是shell的语句啊,所以shell的语句当然可以识别shell的变量了~~~