python考勤分析
上次的ruby考勤那个, 因为天天发邮件, 人事觉得烦, 刚好现在在折腾Python, 于是用这个重写了...
这个发邮件的, 可以发附件,图片... 记录一下
#!/usr/bin/python
import datetime
import MySQLdb
import smtplib
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
from email.mime.image import MIMEImage
csv_str = 'datetime,name,start,over,hour\n'
name = {1:"mei.han", 2:"xing.mei", 3:"you.yao", 6:"zing.ma", 9:"jing.cai"}
csv_file = "/tmp/check_in.csv"
def send_mail(filename=[], picname=[], content_txt='', content_html=''):
smtpserver = 'smtp.163.com'
username = '[email protected]'
password = '123123'
msg = MIMEMultipart()
msg['Subject'] = 'Check_in'
msg['From'] = "[email protected]"
msg['To'] = "[email protected]"
# attchment
if len(filename) > 0:
for i in filename:
att = MIMEText(open(i, 'rb').read(), 'base64', 'gf2312')
att["Content-Type"] = 'application/octet-stream'
att["Content-Disposition"] = 'attachment; filename="%s"' % i.split('/')[-1]
msg.attach(att)
# attchment picture
if len(picname) > 0 and content_html != '':
for i in range(0,len(picname)):
#content_html = '<b>Some <i>HTML</i> text</b> and an image.<br><img src="cid:image%s"><br>good!' % i
msg_content_html = MIMEText(content_html,'html','gb2312')
msg.attach(msg_content_html)
with open(picname[i], 'rb') as f:
msgImage = MIMEImage(f.read())
msgImage.add_header('Content-ID', '<image%s>' % (i + 1))
msg.attach(msgImage)
# content text
if content_txt != '':
msg_content_txt = MIMEText(content_txt,_subtype='plain',_charset='gb2312')
msg.attach(msg_content_txt)
# content html
if content_html != '' and len(picname) == 0:
msg_content_html = MIMEText(content_html,_subtype='html',_charset='gb2312')
msg.attach(msg_content_html)
smtp = smtplib.SMTP()
smtp.connect('smtp.163.com')
smtp.starttls() #ssl send
smtp.login(username, password)
smtp.sendmail(msg['From'], msg['To'], msg.as_string())
smtp.quit()
def check_kq(kq_time):
tmp_list = []
csv_tmp = ''
work = False
try:
conn = MySQLdb.connect(host='192.168.2.2',user='kq',passwd='123123',db='test',port=3306)
cur = conn.cursor()
for i in name.keys():
max_time = kq_time + " 23:00:00"
min_time = kq_time + " 06:00:00"
cur.execute("select * from kqtime where userid=%s and time>\'%s\' and time<\'%s\'" % (i,min_time,max_time))
kqtime = cur.fetchall()
for x,y,z in kqtime: #num, time, userid
tmp_list.append(y)
if len(tmp_list) == 0:
csv_tmp = csv_tmp + ("%s,%s,,,0" % (kq_time,name[i])) + "\n"
elif len(tmp_list) == 1:
if int(tmp_list[0].strftime("%H")) < 12:
csv_tmp = csv_tmp + ("%s,%s,%s,,0" % (kq_time,name[i],tmp_list[0].strftime("%H:%M:%S"))) + "\n"
else:
csv_tmp = csv_tmp + ("%s,%s,,%s,0" % (kq_time,name[i],tmp_list[0].strftime("%H:%M:%S"))) + "\n"
work = True
else:
hour_str = str(max(tmp_list) - min(tmp_list))
hour = int(hour_str.split(':')[0])
sec = int(hour_str.split(':')[1])
if hour > 3:
hour -= 1
if sec < 15:
hour += 0.0
elif 15 <= sec < 45:
hour += 0.5
elif sec >= 45:
hour += 1.0
work = True
csv_tmp = csv_tmp + ("%s,%s,%s,%s,%s" % (kq_time,name[i],min(tmp_list).strftime("%H:%M:%S"),max(tmp_list).strftime("%H:%M:%S"),hour)) + "\n"
tmp_list = []
cur.close()
conn.close()
except:
pass
if work:
return csv_tmp
else:
return ''
for i in range(1,8)[::-1]: #一周分析一次
day_kq = check_kq((datetime.datetime.now() - datetime.timedelta(days=i)).strftime("%Y-%m-%d"))
if day_kq != '':
csv_str = csv_str + check_kq((datetime.datetime.now() - datetime.timedelta(days=i)).strftime("%Y-%m-%d")) + "\n"
with open(csv_file, 'w') as f:
f.write(csv_str)
send_mail(filename=[csv_file], content_txt="Check_In")
os.remove(csv_file)