# -*- coding: UTF-8 -*-
##远程控制电脑并关机

##博客地址:http://www.lylinux.org/python%E8%BF%9C%E7%A8%8B%E7%9B%91%E6%8E%A7%E7%94%B5%E8%84%91%E5%B9%B6%E5%85%B3%E6%9C%BA.html

import time
from VideoCapture import Device
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from email.mime.image import MIMEImage
import smtplib
from PIL import ImageGrab
from PIL import ImageEnhance
import os
from fetion import *
import poplib,email



def get_correcttime() :
 open_time = time.strftime('%m%d%H%M',time.localtime(time.time()))
 return open_time


def get_desktopimg(correcttime):
 filename = r"F:/learn/watchcomputer/desktop" + correcttime + ".jpg"
 pic = ImageGrab.grab()
 pic.save(filename)
 print "desktop img saved ok!!!!"
 return filename
 


def get_webcamimg(correcttime):
 filename = r"F:/learn/watchcomputer/webcam" + correcttime + ".jpg"
 cam = Device()
 
 res = (640,480)
 cam = Device()
 cam.setResolution(res[0],res[1])
 
 brightness = 1.0
 contrast = 1.0
 
 camshot = ImageEnhance.Brightness(cam.getImage()).enhance(brightness)
 camshot = ImageEnhance.Contrast(camshot).enhance(contrast)
 time.sleep(10)
 cam.saveSnapshot(filename,timestamp=3, boldfont=1, quality=80)
 print "webcam img saved ok!!!!!!"
 return filename

def send_img(desktop,webcam):
 try:
 from_mail = "ddddddddd@gmail.com"
 to_mail = "fffffffffffff@gmail.com"
 msg = MIMEMultipart()
 msg['From'] = from_mail
 msg['To'] = to_mail
 msg['Subject'] = 'my computer'
 body = 'test img send'
 html_code = '<b><i>the guy who use my computer and the screen shot.</b></i> <img alt="desktop" src="cid:image1" width="683" height="384"/><img alt="webcam" src="cid:image2"/>'
 print html_code
 con = MIMEText(html_code,'html','utf-8')
 msg.attach(con)
 img1 = MIMEImage(file(desktop,'rb').read())
 img2 = MIMEImage(file(webcam,'rb').read())
 img1.add_header('Content-ID','<image1>')
 img2.add_header('Content-ID','<image2>')
 msg.attach(img1)
 msg.attach(img2)
 server = smtplib.SMTP('smtp.gmail.com')
 server.docmd('ehol',from_mail)
 server.starttls()
 server.login(from_mail,'wwwwwwwwwwwwwwwwwwwwwwwwww')
 server.sendmail(from_mail,to_mail,msg.as_string())
 server.quit()
 print "mail send ok!!!!!"
 return True 
 except:
 return False


def write_log(issuccess,smsissend):
 log_file_dir = r"F:/learn/watchcomputer/computer.log"
 log_flie = file(log_file_dir,'a')
 log_flie.write('at '+now_time + ' img saved success! ')
 if (issuccess):
 log_flie.write('mail send success!!!')
 elif(not issuccess):
 log_flie.write('mail send failed!!!')
 if (smsIsSend):
 log_flie.write('sms send success')
 elif(not smsIsSend):
 log_flie.write('sms send failed')
 log_flie.write('\n\n')
 log_flie.close()



def send_sms(correct_time):
 myphone = '11111111'
 mypwd = '2222222222222'
 tophone = '33333333333333333'
 
 
 print 'send to '+tophone
 sms = 'computer is open at '+correct_time
 try:
 fetion = PyFetion(myphone,mypwd,'TCP')
 fetion.login(FetionHidden) 
 fetion.send_sms(sms,tophone,True)
 fetion.logout()
 return True
 except:
 return False




def accpmail():
 try:
 p=poplib.POP3_SSL('pop.gmail.com')
 p.user("recent:wwwwwwwwwwwwwwwww@gmail.com")
 p.pass_('ffffffffffffffffffffffff')
 
 ret = p.stat()
 except poplib.error_proto,e:
 print "Login failed:",e
 return "fail"
 print "Login succeeded"
 mailnum=ret[0]
 down=p.retr(mailnum)
 for i in down[1]:
 if i.find("Subject:")==0:
 cmdstr=i
 print cmdstr
 break
 p.quit()
 return cmdstr


time.sleep(240) 
now_time = get_correcttime()
desktopimg = get_desktopimg(now_time)
webcamimg = get_webcamimg(now_time)

smsIsSend = send_sms(now_time)


if(os.path.isfile(desktopimg) and os.path.isfile(webcamimg)): 
 MailIsSend = send_img(desktopimg,webcamimg)
 write_log(MailIsSend,smsIsSend)

while True:
 cmdstr = accpmail()
 if cmdstr != "fail":
 if cmdstr.find('shutdown')!=-1:
 os.system("123.mp3")
 elif cmdstr.find("quit") != -1:
 break
 elif cmdstr.find("continue") != -1:
 time.sleep(120)
 continue
 else:
 time.sleep(120)
 continue