모방 창의 Python 버전 예약 작업 tool_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)
    
    # 创建'res/command'文件夹
    if os.path.exists('res/command'):
      pass
    else:
      os.mkdir('res/command')
    
#     self.startDate.textFromDateTime()
  
  
  @pyqtSignature("")
  def on_run_btn_clicked(self):
    """
            创建任务.
    """   
    
    # 在查询栏显示的内容
    cmd = self.taskRun.toPlainText()
    # 构建'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&#39;日期出错&#39;, u&#39;结束日期要大于开始日期&#39;)
    
    elif not p.isdigit():
      QtUtil.showOkDialog(self, u&#39;频率出错&#39;, u&#39;运行频率必须为整数&#39;)
      
    else:
      # 如果任务条件不完整，则创建失败
      if self.taskName.text()==&#39;&#39; or self.modifier.text()==&#39;&#39; or cmd==&#39;&#39;:
        QtUtil.showOkDialog(self, u&#39;创建失败&#39;, u&#39;任务内容缺失&#39;)
      else:
        # 读取&#39;res/command&#39;下所有文件的文件名到filename
        for root, dirs, files in os.walk(&#39;res/command&#39;):
          for file in files:
            filename = open(&#39;res/filename.txt&#39;,&#39;a&#39;)
            filename.write(&#39;/&#39;)
  #           filename.truncate()
            filename.write(str(file))
            filename.close()
            filename1 = open(&#39;res/filename.txt&#39;)
            fn = filename1.read()
            filename1.close()
        print fn
        print &#39;/&#39; + str(self.taskName.text()) + &#39;.cmd&#39;
        # 如果任务名在filename中能找到，则说明任务已经存在
        if &#39;/&#39; + str(self.taskName.text()) + &#39;.cmd&#39; in fn:
          QtUtil.showOkDialog(self, u&#39;创建失败&#39;, u&#39;任务已存在&#39;)
           
        else:
          
          # 任务内容
          if self.schedule.currentText() == &#39;monthly&#39;:
            print &#39;monthly&#39;
            run = &#39;schtasks /create /tn &#39;+ self.taskName.text() + &#39; /tr &#39;+os.getcwd()+&#39;/res/command/&#39;+self.taskName.text()+&#39;.bat /sc &#39; + self.schedule.currentText() + &#39; /d &#39; +self.modifier.text() + &#39; /m &#39; + self.month.currentText() + &#39; /st &#39; + self.timeEdit.text() + &#39; /sd &#39; + self.startDate.text() + &#39; /ed &#39; + self.endDate.text()
          elif self.schedule.currentText() == &#39;once&#39;:
            print &#39;once&#39;
            run = &#39;schtasks /create /tn &#39;+ self.taskName.text() + &#39; /tr &#39;+os.getcwd()+&#39;/res/command/&#39;+self.taskName.text()+&#39;.bat /sc &#39; + self.schedule.currentText() + &#39; /st &#39; + self.timeEdit.text() + &#39; /sd &#39; + self.startDate.text() 
            if self.startDate.text() < time.strftime(&#39;%Y/%m/%d&#39;) or (self.timeEdit.text() <= time.strftime(&#39;%H:%M:%S&#39;) and self.startDate.text() == time.strftime(&#39;%Y/%m/%d&#39;)) :
              QtUtil.showOkDialog(self, u&#39;时间错误&#39;, u&#39;设置时间早于当前时间&#39;)
              return
          else: 
            print &#39;not monthly&#39;
            run = &#39;schtasks /create /tn &#39;+ self.taskName.text() + &#39; /tr &#39;+os.getcwd()+&#39;/res/command/&#39;+self.taskName.text()+&#39;.bat /sc &#39; + self.schedule.currentText() + &#39; /mo &#39; +self.modifier.text() + &#39; /st &#39; + self.timeEdit.text() + &#39; /sd &#39; + self.startDate.text() + &#39; /ed &#39; + self.endDate.text()
          
          # 创建命令文件
          fd = open(&#39;res/command/&#39;+self.taskName.text()+&#39;.bat&#39;,&#39;w&#39;)
          fd.write(cmd)
          fd.close()
          
          # 创建任务文件
          f = open(&#39;res/command/&#39;+self.taskName.text()+&#39;.cmd&#39;,&#39;w&#39;)
          f.write(run)
          f.close()
          # 创建任务
          os.system(os.getcwd()+&#39;\\res\command\\&#39;+str(self.taskName.text())+&#39;.cmd&#39;)
          QtUtil.showOkDialog(self, u&#39;创建成功&#39;, u&#39;创建成功&#39;)
  @pyqtSignature("")
  def on_delete_btn_clicked(self):
    """
            删除任务.
    """
    # 强制删除删除框内任务
    x=os.system(&#39;schtasks /delete /tn &#39;+str(self.taskDelete.text()).decode(&#39;gbk&#39;)+&#39; /f&#39;)
    # 如果已经任务已经删除，则报任务不存在
    if x==1:
      QtUtil.showOkDialog(self, u&#39;删除失败&#39;, u&#39;任务名错误或不存在该任务&#39;)
    else:
      os.remove(&#39;res/filename.txt&#39;)
      
      
      if os.path.exists(&#39;res/command/&#39;+self.taskDelete.text()+&#39;.cmd&#39;):
        os.remove(&#39;res/command/&#39;+str(self.taskDelete.text())+&#39;.bat&#39;)
        os.remove(&#39;res/command/&#39;+str(self.taskDelete.text())+&#39;.cmd&#39;)
        
      # 读取&#39;res/command&#39;下所有文件的文件名到filename
      for root, dirs, files in os.walk(&#39;res/command&#39;):
        for file in files:
          filename = open(&#39;res/filename.txt&#39;,&#39;a&#39;)
          filename.write(&#39;/&#39;)
#           filename.truncate()
          filename.write(str(file))
          filename.close()
          filename1 = open(&#39;res/filename.txt&#39;)
          fn = filename1.read()
          filename1.close()
      # 删除任务，并删除命令文件与任务文件
      QtUtil.showOkDialog(self, u&#39;删除成功&#39;, u&#39;删除成功&#39;) 
      
  @pyqtSignature("")
  def on_query_btn_clicked(self):
    """
            查询任务.
    """
    # 调整 936 为 437 美国编码，才可运行
    os.system(&#39;chcp 437&#39;)
    # 查询任务
    os.system(&#39;schtasks /query&#39;)
    # 在生成新log文件前先删除以前的log文件
    if os.path.exists(&#39;res/log.txt&#39;):
      os.remove(&#39;res/log.txt&#39;)
    
    # 遍历&#39;res/command&#39;的所有文件，将所有文件内容复制到log文件中
    for root, dirs, files in os.walk(&#39;res/command&#39;):
      for file in files:
        dir = str(root)+&#39;/&#39;+str(file)
        f = open(dir,&#39;r&#39;)
        scripts = f.read()
        new_path_filename = &#39;res/log.txt&#39;
        
        f = open(new_path_filename, &#39;a&#39;)
        f.write(scripts) 
        f.write(&#39;\n&#39;)
        f.close()
    # 读取log文件
    if os.path.exists(&#39;res/log.txt&#39;):
      fd=open(&#39;res/log.txt&#39;)
      info = fd.read()
      fd.close()
      # 在查询窗口显示log文件内容
      self.taskQuery.setText(str(info))
    else:
      QtUtil.showOkDialog(self, u&#39;失败&#39;, u&#39;不存在任务&#39;)
  
  @pyqtSignature("")
  def on_delall_btn_clicked(self):
    """
            清空任务.
    """
    os.system(&#39;schtasks /delete /tn * /f&#39;)
    if os.path.exists(&#39;res/log.txt&#39;):
      os.remove(&#39;res/log.txt&#39;)
    if os.path.exists(&#39;res/filename.txt&#39;):  
      os.remove(&#39;res/filename.txt&#39;)
    shutil.rmtree(&#39;res/command&#39;)
    os.mkdir(&#39;res/command&#39;)
    QtUtil.showOkDialog(self, u&#39;成功&#39;, u&#39;任务清空&#39;)