size = 0

nums = 0

for root, dirs, files in os.walk(dir_path):

size += sum([os.path.getsize(os.path.join(root, name)) for name in files])

nums += len(files)

port = int(port)

sk = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

sk.settimeout(timeout)

result = True

try:

sk.connect((ip, port))

except Exception:

result = False

sk.close()

address = cfg.read('check_server_ip')

port = int(cfg.read('check_server_port'))

time_out = float(cfg.read('check_server_time_out'))

if check_port(port, address, time_out):

return True

def __init__(self):

# Try to open database file, if failed then create a new one

try:

open(cfg.read('db_filename'), 'r').close()

except:

self.InitDataBase()

sys.stderr.write('Warning: database not found, created a new one!\n')

self.mutex = Lock()

# Method to create a entire new database

def InitDataBase(self):

open(cfg.read('db_filename'), 'w').close()

with open(cfg.read('sql_filename')) as fid,\

sqlite3.connect(cfg.read('db_filename')) as conn:

conn.text_factory = str

for sql in fid.read().split('\n\n'):

if len(sql) > 3:

conn.execute(sql)

conn.commit()

def Execute(self, sql):

self.mutex.acquire()

# connect to the database

with sqlite3.connect(cfg.read('db_filename')) as conn:

conn.text_factory = str

conn.execute(sql)

conn.commit()

self.mutex.release()

def Query(self, sql):

with sqlite3.connect(cfg.read('db_filename')) as conn:

conn.text_factory = str

cursor = conn.execute(sql)

return cursor.fetchall()

def QueryFirst(self, sql):

with sqlite3.connect(cfg.read('db_filename')) as conn:

conn.text_factory = str

cursor = conn.execute(sql)

def __init__(self):

try:

open(cfg_name, 'r').close()

except:

sys.stderr.write(u'严重错误，无法读取配置文件！程序自动退出。\n')

exit(-1)

self.config = ConfigParser()

self.config.read(cfg_name)

self.system = dict(self.config.items('system'))

self.config_check()

def config_check(self):

try:

# 首先检查设定的全局存储目录是否合法

check_path(self.read('global_pos'))

# 然后检查管理员的下载目录是否存在

root_path = os.path.join(self.read('global_pos'), 'root')

if not os.path.exists(root_path):

os.mkdir(root_path)

except Exception, err:

sys.stderr.write(u'系统错误，原因：%s\n' % err)

exit(-1)

# 接下来检查端口是否可用

if check_port(self.read('port_name')):

sys.stderr.write(u'系统错误，端口被占用！\n')

exit(-1)

def read(self, key):

result = self.system.get(key)

if type(result) == str:

result = result.decode('gb18030')

return result

def write(self, key, value):

if type(value) == unicode:

value = value.encode('gb18030')

self.system[key] = value

self.config.set('system', key, value)

def __init__(self):

open(cfg.read('log_filename'), 'w').close()

self.mutex = Lock()

def log_message(self, msg):

self.mutex.acquire()

with open(cfg.read('log_filename'), 'a') as fid:

# 自动加上换行

if msg[-1] != '\n':

msg += '\n'

if type(msg) == unicode:

msg = msg.encode('gb18030')

fid.write(msg)