Python实现简单的文件传输与MySQL备份的脚本分享

用python实现简单Server/Client文件传输：

服务器端:

#!/usr/bin/python
import SocketServer, time
class MyServer(SocketServer.BaseRequestHandler):
userInfo = {
\'leonis\' : \'leonis\',
\'hudeyong\' : \'hudeyong\',
\'mudan\' : \'mudan\' }
def handle(self):
print \'Connected from\', self.client_address
while True:
receivedData = self.request.recv(8192)
if not receivedData:
continue
elif receivedData == \'Hi, server\':
self.request.sendall(\'hi, client\')
elif receivedData.startswith(\'name\'):
self.clientName = receivedData.split(\':\')[-1]
if MyServer.userInfo.has_key(self.clientName):
self.request.sendall(\'valid\')
else:
self.request.sendall(\'invalid\')
elif receivedData.startswith(\'pwd\'):
self.clientPwd = receivedData.split(\':\')[-1]
if self.clientPwd == MyServer.userInfo[self.clientName]:
self.request.sendall(\'valid\')
time.sleep(5)
sfile = open(\'down.sh\', \'rb\')
while True:
data = sfile.read(1024)
if not data:
break
while len(data) > 0:
intSent = self.request.send(data)
data = data[intSent:]
time.sleep(3)
self.request.sendall(\'EOF\')
else:
self.request.sendall(\'invalid\')
elif receivedData == \'bye\':
break
self.request.close()
print \'Disconnected from\', self.client_address
print
if __name__ == \'__main__\':
print \'Server is started\\nwaiting for connection…\\n\'
srv = SocketServer.ThreadingTCPServer((\'ip\', 50000), MyServer)
srv.serve_forever()

客户端:

import socket, time
class MyClient:
def __init__(self):
print \'Prepare for connecting…\'
def connect(self):
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect((\'ip\', 50000))
sock.sendall(\'Hi, server\')
self.response = sock.recv(8192)
print \'Server:\', self.response
self.s = raw_input(\"Server: Do you want get the \'thinking in python\' file?(y/n):\")
if self.s == \'y\':
while True:
self.name = raw_input(\'Server: input our name:\')
sock.sendall(\'name:\' + self.name.strip())
self.response = sock.recv(8192)
if self.response == \'valid\':
break
else:
print \'Server: Invalid username\'
while True:
self.pwd = raw_input(\'Server: input our password:\')
sock.sendall(\'pwd:\' + self.pwd.strip())
self.response = sock.recv(8192)
if self.response == \'valid\':
print \'please wait…\'
f = open(\'down.sh\', \'wb\')
while True:
data = sock.recv(1024)
if data == \'EOF\':
break
f.write(data)
f.flush()
f.close()
print \'download finished\'
break
else:
print \'Server: Invalid password\'
sock.sendall(\'bye\')
sock.close()
print \'Disconnected\'
if __name__ == \'__main__\':
client = MyClient()
client.connect()

由于担心服务器数据安全,所以写了这个脚本,结合上面分享的Server/Client 文件互传,可以备份网站数据到本地,安全又可靠

#!/usr/bin/python
# Filename: webbak.py
import os
import time
import tarfile
 
os.chdir(\'/home/web/\') #切换目录
source = \'leonis\'
bakdir = \'/home/web/leonis/\'
# mysql dump
dump = \'mysqldump\'
dbuser = \'XXXXXXX\'
dbpwd = \'XXXXXXXXXXX\'
dbname = \'XXXXXXXX\'
sqlfile = \'/home/web/leonis/leonis.sql\'
sql = \"%s -u%s -p%s %s > %s\" % (dump,dbuser,dbpwd,dbname,sqlfile)
if os.path.exists(sqlfile): 
  os.remove(sqlfile)
else:
  print \'then will dump sql file\'
result = os.popen(sql)
if result:       #
 print (\"SQL backup completed!\")
else:
 print (\"SQL backup failed!\")
# gzip 压缩 以当日日期命名
filename = bakdir + time.strftime(\'%Y%m%d\')+\'.tar.gz\'
tar = tarfile.open(filename,\"w:gz\")
tar.add(source)
tar.close()