打算写一个备份的程序

bg7nyt

在网上找了一下，没有找到比较好的备份脚本，打算写一个备份的程序。




还没写完，等有时间再写，
请关注：http://netkiller.github.com/architect/solution/solution.backup.html

1. #!/usr/bin/env python3
   
2. #/bin/env python3
   
3. #-*- coding: utf-8 -*-
   
4. ##############################################
   
5. # Home  : http://netkiller.sf.net
   
6. # Author: Neo <openunix@163.com>
   
7. ##############################################
   
8. 
   
9. try:
   
10.     import logging, configparser
    
11.     import threading
    
12.     from optparse import OptionParser, OptionGroup
    
13.     import os,io,sys
    
14. except ImportError as err:
    
15.     print("Error: %s" %(err))
    
16. 
    
17. class Runtime(threading.Thread):
    
18.     def __init__(self, logging):
    
19.         threading.Thread.__init__(self)
    
20.         self.logging = logging
    
21.         sys.stdin = open('stdin.log', 'r')
    
22.         sys.stdout = open('stdout.log', 'w')
    
23.         sys.stderr = open('stderr.log', 'w')
    
24.     def command(self,cmd):
    
25.         commands = {
    
26.             'rsync': 'rsync -auzvP',
    
27.             'sftp' : 'sftp',
    
28.             'scp' : 'scp -r',
    
29.             'cp' : 'cp -r'
    
30.         }
    
31.         return commands[cmd]
    
32.     def policy(self,policy):
    
33.         policies = {
    
34.             'full': 'rsync -az',
    
35.             'mirror': 'rsync -auz --delete',
    
36.             'differential': 'rsync -auz --delete',
    
37.             'incremental': 'rsync -auz',
    
38.             'clone': 'dd',
    
39.             'copy.cp': 'cp -au',
    
40.             'copy.cp.backup': "cp  --suffix=$(date '+.%Y-%m-%d.%H:%M:%S') "
    
41.             'copy.scp': 'scp -a',
    
42.             'mirror.ftp': 'wget -m',
    
43.             'mirror.http': 'wget -m',
    
44.             'archive.zip': 'zip',
    
45.             'archive.7zip': '7zip',
    
46.             'archive.cpio': 'cpio',
    
47.             'archive.tar': 'tar zcvf',
    
48.             'snapshot': ''
    
49.         }
    
50.         return policies[policy]
    
51.     def execute(self,cfg):
    
52.         print( self.policy('mirror'))
    
53.         #command = self.command(cfg['cmd'])
    
54.         command = self.policy(cfg['policy']) + ' ' + cfg['source'] + ' ' + cfg['target'] + ' >> stdout.log'
    
55.         self.logging.debug(command)
    
56.         os.system(command)
    
57. class Task():
    
58.     def __init__(self, logging):
    
59.         self.logging = logging
    
60.         self.config = configparser.SafeConfigParser()
    
61.         cfg='task.cfg'
    
62.         self.config.read(cfg)
    
63.     def list(self):
    
64.         for section in self.config.sections():
    
65.             print(section)
    
66.     def new(self):
    
67.         pass
    
68.     def remove(self):
    
69.         pass
    
70.     def change(self):
    
71.         pass
    
72.     def run(self, section):
    
73.         try:
    
74.             #print(cfg)
    
75.             cfg = self.config.items(section)
    
76.             r = Runtime(self.logging)
    
77.             r.execute(dict(cfg))
    
78.         except configparser.NoSectionError as err:
    
79.             print(err)
    
80.     def show(self, section):
    
81.         for item in self.config.items(section):
    
82.             #k,v = item
    
83.             print("%s: %s" %(item))
    
84.     def get(self,section):
    
85.         return self.config.items(section)
    
86. class Schedule():
    
87.     def __init__(self, logging):
    
88.         self.logging = logging
    
89.         self.config = configparser.SafeConfigParser()
    
90.         cfg='schedule.cfg'
    
91.         self.config.read(cfg)
    
92.     def list(self):
    
93.         for section in self.config.sections():
    
94.             print(section)
    
95.     def show(self, section):
    
96.         for item in self.config.items(section):
    
97.             print("%s: %s" %(item))
    
98.     def new(self):
    
99.         pass
    
100.     def remove(self):
     
101.         pass
     
102.     def change(self):
     
103.         system('backup.cron')
     
104.     def status(self):
     
105.         pass
     
106.     def run(self, section):
     
107.         threads = []
     
108.         t = Task(self.logging)
     
109.         #t.run(task)
     
110.         for task,status in self.config.items(section):
     
111.             if status :
     
112.                 cfg = t.get(task)
     
113.                 r  = Runtime(dict(cfg))
     
114.                 r.setName('Thread-' + task)
     
115.                 threads.append(r)
     
116.         for t in threads:
     
117.             #print(t.getName())
     
118.             self.logging.info(t.getName())
     
119.             t.start()
     
120.             t.join()
     
121. class Volume():
     
122.     pass
     
123. class Backup():
     
124.     def __init__(self):
     
125.         self.config = {}
     
126.         #self.config['logfile'] = 'backup.log'
     
127. 
     
128.         usage = "usage: %prog [options] arg1 arg2"
     
129.         self.parser = OptionParser(usage)
     
130.         self.parser.add_option("-f", "--file", dest="filename",
     
131.                   help="write report to FILE", metavar="FILE")
     
132.         self.parser.add_option("-q", "--quiet",
     
133.                   action="store_false", dest="verbose", default=True,
     
134.                   help="don't print status messages to stdout")
     
135.         self.parser.add_option('','--config', dest="config", help='Read configuration options from file.', default='backup.cfg')
     
136. 
     
137.         group = OptionGroup(self.parser, "arg1",
     
138.                     "arg1: task | schedule")
     
139.         self.parser.add_option_group(group)
     
140.         group = OptionGroup(self.parser, 'arg2', 'arg2: list | new | remove | show')
     
141.         self.parser.add_option_group(group)
     
142. 
     
143.         self.parser.add_option('-v','--version',action='store_true', help='print version number')
     
144.         self.parser.add_option('-d','--daemon', dest='daemon', action='store_true', help='run as daemon')
     
145.         self.parser.add_option('','--logfile', help='logs file.', default='backup.log')
     
146. 
     
147.         (options, args) = self.parser.parse_args()
     
148.         self.configure(options)
     
149. 
     
150.         try:
     
151.             logging.basicConfig(level=logging.NOTSET,
     
152.                     format='%(asctime)s %(levelname)-8s %(message)s',
     
153.                     datefmt='%Y-%m-%d %H:%M:%S',
     
154.                     filename=self.config['environment']['logfile'],
     
155.                     filemode='a')
     
156.             self.logging = logging.getLogger()
     
157.         except AttributeError as err:
     
158.             print("Error: %s %s" %(err, self.config['environment']['logfile']))
     
159.             sys.exit(2)
     
160.         pass
     
161. 
     
162.     def configure(self,options):
     
163.         if options.config:
     
164.             cpr = configparser.SafeConfigParser()
     
165.             cpr.read(options.config)
     
166.             for sect in cpr.sections():
     
167.                 self.config[sect] = dict(cpr.items(sect))
     
168.                 #for (key,value) in cpr.items(sect):
     
169.                 #    self.config[key] = value
     
170.         #print(self.config)
     
171. 
     
172.     def task(self, args):
     
173.         try:
     
174.             t = Task(self.logging)
     
175.             if len(args) <= 1:
     
176.                 t.list()
     
177.             elif args[1] == 'list':
     
178.                 t.list()
     
179.             elif args[1] == 'run':
     
180.                 if len(args) == 3:
     
181.                     t.run(args[2])
     
182.                 else:
     
183.                     t.list()
     
184.             elif args[1] == 'show':
     
185.                 if len(args) == 3:
     
186.                     t.show(args[2])
     
187.                 else:
     
188.                     t.list()
     
189.             else:
     
190.                 t.list()
     
191.         except IOError as err:
     
192.             print(err)
     
193.         except configparser.NoSectionError as err:
     
194.             t.list()
     
195.             print(err)
     
196.             self.logging.error(err)
     
197.     def schedule(self,args):
     
198.         try:
     
199.             s = Schedule(self.logging)
     
200.             if len(args) <= 1:
     
201.                 s.list()
     
202.             elif args[1] == 'list':
     
203.                 s.list()
     
204.             elif args[1] == 'show':
     
205.                 if len(args) == 3:
     
206.                     s.show(args[2])
     
207.                 else:
     
208.                     s.list()
     
209.             elif args[1] == 'run':
     
210.                 if len(args) == 3:
     
211.                     s.run(args[2])
     
212.                 else:
     
213.                     s.list()
     
214.             else:
     
215.                 s.list()
     
216.         except configparser.NoSectionError as err:
     
217.             s.list()
     
218.             self.logging.error(err)
     
219.             print(err)
     
220.     def usage(self):
     
221.         self.parser.print_help()
     
222. 
     
223.     def main(self):
     
224.         (options, args) = self.parser.parse_args()
     
225.         if options.daemon:
     
226.             pid = os.fork()
     
227.             if pid > 0:
     
228.                 self.logging.info('daemon is ok')
     
229.                 sys.exit(0)
     
230.         if not args:
     
231.             self.usage()
     
232.         elif args[0] == 'task':
     
233.             self.task(args)
     
234.             self.logging.debug('Task')
     
235.         elif args[0] == 'schedule':
     
236.             self.schedule(args)
     
237.             self.logging.debug('Schedule')
     
238.         else:
     
239.             print('')
     
240. if __name__ == '__main__':
     
241.     try:
     
242.         backup = Backup()
     
243.         backup.main()
     
244.     except KeyboardInterrupt:
     
245.         print ("Crtl+C Pressed. Shutting down.")
     

复制代码

chenyx

mark,楼主写的不错,关注下.