add codescripts

coolcoding · coolcoding · commit a2b6d7dc3383 · 2014-11-01T07:41:53.000+08:00
diff --git a/examples/codeScripts/backup_wp_site.py b/examples/codeScripts/backup_wp_site.py
@@ -0,0 +1,129 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+'''
+	wordpress 备份脚本 
+        备份网站，添加apache伪静态支持.htaccess,备份数据库
+        网站备份位置:/data/backup/wordpress_backup
+        #备份全部wordpress网站
+        python  backup_wp_site.py
+        #备份指定wordpress网站
+        python backup_wp_site.py -f filename(每行一网站 www开头)
+'''
+import os
+import shutil
+import sys
+from string import Template
+import re
+from  optparse  import OptionParser
+
+
+wp_conf_path = '/usr/local/nginx/conf/vhosts/wordpress'
+backup_site = '/data/backup/wordpress_backup'
+wp_path = '/data/wwwroot/wordpress'
+dbrootpass = '8pIi5crkPpUbLazs'
+
+
+
+with open('htaccess','r') as htf:
+    ht_con = htf.read()
+
+
+
+
+def writeHtaccess(path,d=dict()):
+    global ht_con
+    con = Template(ht_con)    
+    con = con.safe_substitute(d)
+    with open(path,'w+') as ff:
+        ff.write(con)
+    
+def is_exist(path):
+    if os.path.exists(path):
+        return path
+    else:
+        print '%s has not found!' % path
+        sys.exit(1)
+
+
+def getDbInfo(path):
+    
+    '''from config get db info'''
+
+    if os.path.isfile(path):
+        with open(path) as f:
+            con = f.read()      
+        user_pattern = r'\s+fastcgi_param USERNAME\s+(.*);'
+        user_pattern_obj = re.compile(user_pattern)
+        dbuser = user_pattern_obj.findall(con)[0]
+        db_pattern = r'\s+fastcgi_param DATABASE\s+(.*);'
+        db_pattern_obj = re.compile(db_pattern)
+        dbname = db_pattern_obj.findall(con)[0]
+        pass_pattern = r'\s+fastcgi_param PASSWORD\s+(.*);'
+        pass_pattern_obj = re.compile(pass_pattern)
+        dbpass = pass_pattern_obj.findall(con)[0]
+        return [dbname,dbuser,dbpass]
+    return None
+        
+
+
+def backup_one_site(site):
+
+    '''backup one site'''
+
+    if site.startswith('www'):
+        site_src = is_exist(os.path.join(wp_path,site + '/webroot'))
+        site_dest = os.path.join(backup_site,site)
+        if not os.path.isdir(site_dest):
+            shutil.copytree(site_src,site_dest)
+        ht_site_dest = os.path.join(site_dest,'.htaccess')
+        writeHtaccess(ht_site_dest,dict(host=site[4:]))
+        conf_src = is_exist(os.path.join(wp_conf_path,site + '.conf'))
+        info =  getDbInfo(conf_src)
+        if not info:
+            print 'database infomation not found'
+            sys.exit(1)
+        print info
+        dump_command  = 'mysqldump -uroot -p%s %s > %s' % (dbrootpass,info[0],os.path.join(site_dest,info[0]+'.sql'))
+        os.system(dump_command)
+        wp_list = '%s | %s\n' %  (site,' | '.join(info))
+        with open('wp_list_info','w+') as wpinfo:
+            wpinfo.write(wp_list)
+ 
+
+    
+def backup_all():
+
+    '''backup all wordpress site'''
+
+    for site in os.listdir(wp_path):
+        backup_one_site(site)
+        sys.exit(1)
+
+
+def walk_backup_site(infile):
+
+    '''backup site in file infile'''
+
+    if not os.path.isfile(infile):
+        print 'file %s not found!' % infile
+        sys.exit(1)
+
+    for site in file(infile):
+        site = site.strip()
+        backup_one_site(site)
+
+
+if __name__ == '__main__':
+    print __doc__
+    usage = "usage: %prog [option]"
+    parse = OptionParser(usage=usage)
+    parse.add_option('-f','--file',dest='filename',default='all',help='read data from filename')
+    parse.add_option('-p','--password',dest='passwd',help='mysql root password')
+    (opt,args) = parse.parse_args()
+    if opt.passwd:
+        dbrootpass = opt.passwd
+    if not opt.filename == 'all':
+        walk_backup_site(opt.filename)
+    else:
+        backup_all()
diff --git a/webapp/requirement.txt b/webapp/requirement.txt
@@ -0,0 +1,3 @@
+mysql-connector-python
+jinja2
+MySQL-python
diff --git a/webapp/www/transwarp/db.py b/webapp/www/transwarp/db.py
@@ -159,8 +159,9 @@ def delete(table,**kw):
     sql = 'delete from `%s` where %s' % (table,where)
     return _update(sql,*args)
 
-@with_connection
-def select(sql,first,*args):
+
+
+def _select(sql,first,*args):
 
     global _db_ctx
     cursor = None
@@ -172,7 +173,6 @@ def select(sql,first,*args):
         names =[x[0] for x in cursor.description] 
         if first:
             values = cursor.fetchone()
-            print values
             if not values:
                 return None
             return Dict(names,values)
@@ -182,7 +182,29 @@ def select(sql,first,*args):
             cursor.close()
 
     
+@with_connection
+def select_one(sql,*args):
+    '''
+    select_one('select * from user where id = ?','1')
+    '''
+    return _select(sql,True,*args)
+
 
+@with_connection    
+def select_int(sql,*args):
+    
+    d = _select(sql,True,*args)
+    if len(d) != 1:
+        raise MuiltiColumnsError('Expect only one column.')
+    return d.values()[0]
+
+
+@with_connection
+def select(sql,*args):
+    '''
+    select('select * from user where id = ?','1')
+    '''
+    return _select(sql,False,*args)
 
 
 if __name__ == '__main__':
@@ -194,4 +216,5 @@ def select(sql,first,*args):
 #    print insert('user',id='2',name='lo')
 #    print update('update user set name = ? where id = ?','good','1')
 #    print delete('user',id='1')
-    print select('select * from user',False)
+    print select_one('select * from user where id = ?','1')
+    print select('select * from user where id = ?','1')
diff --git a/webapp/www/transwarp/orm.py b/webapp/www/transwarp/orm.py
@@ -0,0 +1,31 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+__author__ = 'coolcoding'
+
+import time,logging
+
+class Model(dict):
+
+
+    def __init__(self,**kw):
+
+        super(Model,self).__init__(**kw)
+        
+
+    def __getattr__(self,key):
+        try:
+            return self[key]
+        except KeyError:
+            raise AttributeError(r"'Dict' object has no attribute '%s'" % key)
+
+
+    def __setattr__(self,key,value):
+        
+        self[key] = value
+
+
+
+if __name__ == '__main__':
+    m = Model()
+    print m.user

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+mysql-connector-python`
	`2`	`+jinja2`
	`3`	`+MySQL-python`