python版本的仿windows計(jì)劃任務(wù)工具_(dá)python

# -*- coding: utf-8 -*-

"""
Module implementing App.
"""

from PyQt4.QtGui import QMainWindow
from PyQt4.QtCore import pyqtSignature
import time,os
import QtUtil
import shutil
import time

from v.Ui_App import Ui_MainWindow

class App(QMainWindow, Ui_MainWindow):
 """
 Class documentation goes here.
 """
 def __init__(self, parent = None):
 """
 Constructor
 """
 QMainWindow.__init__(self, parent)
 self.setupUi(self)
 
 # 創(chuàng)建'res/command'文件夾
 if os.path.exists('res/command'):
 pass
 else:
 os.mkdir('res/command')
 
# self.startDate.textFromDateTime()
 
 
 @pyqtSignature("")
 def on_run_btn_clicked(self):
 """
 創(chuàng)建任務(wù).
 """ 
 
 # 在查詢欄顯示的內(nèi)容
 cmd = self.taskRun.toPlainText()
 # 構(gòu)建'res/command/01.txt'文件
 if not os.path.exists('res/command/01.txt'):
 m = open('res/command/01.txt','w')
 m.close()
 p = str(self.modifier.text())
 # 判斷日期是否正確
 if self.endDate.text() <= self.startDate.text():
 QtUtil.showOkDialog(self, u'日期出錯(cuò)', u'結(jié)束日期要大于開始日期')
 
 elif not p.isdigit():
 QtUtil.showOkDialog(self, u'頻率出錯(cuò)', u'運(yùn)行頻率必須為整數(shù)')
 
 else:
 # 如果任務(wù)條件不完整，則創(chuàng)建失敗
 if self.taskName.text()=='' or self.modifier.text()=='' or cmd=='':
 QtUtil.showOkDialog(self, u'創(chuàng)建失敗', u'任務(wù)內(nèi)容缺失')
 else:
 # 讀取'res/command'下所有文件的文件名到filename
 for root, dirs, files in os.walk('res/command'):
 for file in files:
 filename = open('res/filename.txt','a')
 filename.write('/')
 # filename.truncate()
 filename.write(str(file))
 filename.close()
 filename1 = open('res/filename.txt')
 fn = filename1.read()
 filename1.close()
 print fn
 print '/' + str(self.taskName.text()) + '.cmd'
 # 如果任務(wù)名在filename中能找到，則說明任務(wù)已經(jīng)存在
 if '/' + str(self.taskName.text()) + '.cmd' in fn:
 QtUtil.showOkDialog(self, u'創(chuàng)建失敗', u'任務(wù)已存在')
 
 else:
 
 # 任務(wù)內(nèi)容
 if self.schedule.currentText() == 'monthly':
 print 'monthly'
 run = 'schtasks /create /tn '+ self.taskName.text() + ' /tr '+os.getcwd()+'/res/command/'+self.taskName.text()+'.bat /sc ' + self.schedule.currentText() + ' /d ' +self.modifier.text() + ' /m ' + self.month.currentText() + ' /st ' + self.timeEdit.text() + ' /sd ' + self.startDate.text() + ' /ed ' + self.endDate.text()
 elif self.schedule.currentText() == 'once':
 print 'once'
 run = 'schtasks /create /tn '+ self.taskName.text() + ' /tr '+os.getcwd()+'/res/command/'+self.taskName.text()+'.bat /sc ' + self.schedule.currentText() + ' /st ' + self.timeEdit.text() + ' /sd ' + self.startDate.text() 
 if self.startDate.text() < time.strftime('%Y/%m/%d') or (self.timeEdit.text() <= time.strftime('%H:%M:%S') and self.startDate.text() == time.strftime('%Y/%m/%d')) :
 QtUtil.showOkDialog(self, u'時(shí)間錯(cuò)誤', u'設(shè)置時(shí)間早于當(dāng)前時(shí)間')
 return
 else: 
 print 'not monthly'
 run = 'schtasks /create /tn '+ self.taskName.text() + ' /tr '+os.getcwd()+'/res/command/'+self.taskName.text()+'.bat /sc ' + self.schedule.currentText() + ' /mo ' +self.modifier.text() + ' /st ' + self.timeEdit.text() + ' /sd ' + self.startDate.text() + ' /ed ' + self.endDate.text()
 
 # 創(chuàng)建命令文件
 fd = open('res/command/'+self.taskName.text()+'.bat','w')
 fd.write(cmd)
 fd.close()
 
 # 創(chuàng)建任務(wù)文件
 f = open('res/command/'+self.taskName.text()+'.cmd','w')
 f.write(run)
 f.close()
 # 創(chuàng)建任務(wù)
 os.system(os.getcwd()+'\rescommand\'+str(self.taskName.text())+'.cmd')
 QtUtil.showOkDialog(self, u'創(chuàng)建成功', u'創(chuàng)建成功')
 @pyqtSignature("")
 def on_delete_btn_clicked(self):
 """
 刪除任務(wù).
 """
 # 強(qiáng)制刪除刪除框內(nèi)任務(wù)
 x=os.system('schtasks /delete /tn '+str(self.taskDelete.text()).decode('gbk')+' /f')
 # 如果已經(jīng)任務(wù)已經(jīng)刪除，則報(bào)任務(wù)不存在
 if x==1:
 QtUtil.showOkDialog(self, u'刪除失敗', u'任務(wù)名錯(cuò)誤或不存在該任務(wù)')
 else:
 os.remove('res/filename.txt')
 
 
 if os.path.exists('res/command/'+self.taskDelete.text()+'.cmd'):
 os.remove('res/command/'+str(self.taskDelete.text())+'.bat')
 os.remove('res/command/'+str(self.taskDelete.text())+'.cmd')
 
 # 讀取'res/command'下所有文件的文件名到filename
 for root, dirs, files in os.walk('res/command'):
 for file in files:
 filename = open('res/filename.txt','a')
 filename.write('/')
# filename.truncate()
 filename.write(str(file))
 filename.close()
 filename1 = open('res/filename.txt')
 fn = filename1.read()
 filename1.close()
 # 刪除任務(wù)，并刪除命令文件與任務(wù)文件
 QtUtil.showOkDialog(self, u'刪除成功', u'刪除成功') 
 
 @pyqtSignature("")
 def on_query_btn_clicked(self):
 """
 查詢?nèi)蝿?wù).
 """
 # 調(diào)整 936 為 437 美國編碼，才可運(yùn)行
 os.system('chcp 437')
 # 查詢?nèi)蝿?wù)
 os.system('schtasks /query')
 # 在生成新log文件前先刪除以前的log文件
 if os.path.exists('res/log.txt'):
 os.remove('res/log.txt')
 
 # 遍歷'res/command'的所有文件，將所有文件內(nèi)容復(fù)制到log文件中
 for root, dirs, files in os.walk('res/command'):
 for file in files:
 dir = str(root)+'/'+str(file)
 f = open(dir,'r')
 scripts = f.read()
 new_path_filename = 'res/log.txt'
 
 f = open(new_path_filename, 'a')
 f.write(scripts) 
 f.write('
')
 f.close()
 # 讀取log文件
 if os.path.exists('res/log.txt'):
 fd=open('res/log.txt')
 info = fd.read()
 fd.close()
 # 在查詢窗口顯示log文件內(nèi)容
 self.taskQuery.setText(str(info))
 else:
 QtUtil.showOkDialog(self, u'失敗', u'不存在任務(wù)')
 
 @pyqtSignature("")
 def on_delall_btn_clicked(self):
 """
 清空任務(wù).
 """
 os.system('schtasks /delete /tn * /f')
 if os.path.exists('res/log.txt'):
 os.remove('res/log.txt')
 if os.path.exists('res/filename.txt'): 
 os.remove('res/filename.txt')
 shutil.rmtree('res/command')
 os.mkdir('res/command')
 QtUtil.showOkDialog(self, u'成功', u'任務(wù)清空')