194 lines
5.7 KiB
Python
194 lines
5.7 KiB
Python
import pymysql as mysql
|
|
import datetime
|
|
from datetime import datetime, timedelta
|
|
import hashlib
|
|
import os
|
|
import sys
|
|
import quest
|
|
|
|
def connect():
|
|
try:
|
|
db=mysql.connect(host='124.70.178.159',user='admin',passwd='Rc5345750.',port=3306)
|
|
print("connect mysql success.")
|
|
return db
|
|
except Exception as e:
|
|
print("can not connect service.")
|
|
return None
|
|
|
|
|
|
# 获取北京时间
|
|
def get_date():
|
|
now_time = datetime.utcnow()
|
|
utc_time = now_time + timedelta(hours=8) # UTC只是比北京时间提前了8个小时
|
|
utc_time = utc_time.strftime("%Y%m%d")
|
|
return utc_time
|
|
|
|
# 获取北京时间
|
|
def get_time():
|
|
now_time = datetime.utcnow()
|
|
utc_time = now_time + timedelta(hours=8) # UTC只是比北京时间提前了8个小时
|
|
utc_time = utc_time.strftime("%Y-%m-%d %H:%M:%S")
|
|
return utc_time
|
|
|
|
|
|
# 获取数据md5
|
|
def md5(data:bytearray):
|
|
m=hashlib.md5(data).hexdigest()
|
|
# print(m)
|
|
return m
|
|
|
|
|
|
# 获得主板sn号
|
|
def get_computer_sn():
|
|
sn = os.popen("wmic bios get serialnumber").readlines()
|
|
return sn[2].strip()
|
|
|
|
|
|
class sql:
|
|
def __init__(self) -> None:
|
|
self.download_path="download/"
|
|
|
|
|
|
# 初始化返回True成功
|
|
def init(self,table_name:str):
|
|
self.db=connect()
|
|
self.table_name=table_name
|
|
if(self.db!=None):
|
|
self.cur=self.db.cursor()
|
|
self.cur.execute("use andy_data")
|
|
self.cur.execute("select version()")
|
|
a=self.cur.fetchone()
|
|
print(a)
|
|
self.create_table(self.table_name)
|
|
return True
|
|
return False
|
|
|
|
# 创建表
|
|
def create_table(self,table_name:str):
|
|
cmd="""CREATE TABLE IF NOT EXISTS `{d1}`(
|
|
`id` INT UNSIGNED AUTO_INCREMENT,
|
|
`time` VARCHAR(30) NOT NULL,
|
|
`name` VARCHAR(256) NOT NULL,
|
|
`md5` VARCHAR(33) NOT NULL,
|
|
`data` MEDIUMBLOB NOT NULL,
|
|
`info` VARCHAR(512),
|
|
PRIMARY KEY ( `id` )
|
|
)ENGINE=InnoDB DEFAULT CHARSET=utf8;""".format(d1=table_name)
|
|
|
|
self.cur.execute(cmd)
|
|
|
|
# 插入数据
|
|
def insert(self,file_name:str,info:str=""):
|
|
s=file_name.split('.')
|
|
if(len(s)<2):
|
|
print("file name without type suffix,will not insert.")
|
|
return
|
|
with open(file_name,"rb") as f:
|
|
d=f.read()
|
|
md=md5(d)
|
|
lis=self.show()
|
|
if(len(lis)>0):
|
|
if(lis[-1][3]==md):
|
|
print("the same file was saved,will not insert.")
|
|
return
|
|
try:
|
|
cmd="INSERT INTO {d1} (time,name,md5,data,info) VALUES (%s,%s,%s,%s,%s);".format(d1=self.table_name)
|
|
self.db.begin()
|
|
self.cur.execute(cmd,([get_time(),file_name,md,d,info]))
|
|
self.db.commit()
|
|
print("insert file success.")
|
|
except Exception as e:
|
|
self.db.rollback()
|
|
print(str(e))
|
|
|
|
# 查看数据
|
|
def show(self):
|
|
cmd= "select id,time,name,md5,info from {d1};".format(d1=self.table_name)
|
|
self.cur.execute(cmd)
|
|
a=self.cur.fetchall()
|
|
# for i in a:
|
|
# print(i[0],i[1],i[2],i[3])
|
|
return a
|
|
|
|
def show_tables(self):
|
|
cmd= "show tables"
|
|
self.cur.execute(cmd)
|
|
a=self.cur.fetchall()
|
|
for i in a:
|
|
print(i)
|
|
return a
|
|
|
|
# 下载文件数据
|
|
def download_data(self,table_name:str,id:int):
|
|
cmd="select name,data from {d1} WHERE id={d2};".format(d1=table_name,d2=id)
|
|
self.cur.execute(cmd)
|
|
a=self.cur.fetchall()[0]
|
|
name=a[0].replace('\\','/').split('/')[-1].split('.')
|
|
name='.'.join(name[:-1])+'--'+str(id)+'.'+name[-1]
|
|
return (name,a[1])
|
|
|
|
# 下载指定文件,返回文件路径
|
|
def download(self,id:int):
|
|
if not os.path.exists(self.download_path):
|
|
os.makedirs(self.download_path)
|
|
ack,name=self.exists(id)
|
|
if(ack==True):
|
|
print("the same file is exists,will not download.")
|
|
return name
|
|
cmd="select name,data from {d1} WHERE id={d2};".format(d1=self.table_name,d2=id)
|
|
self.cur.execute(cmd)
|
|
a=self.cur.fetchall()
|
|
for i in a:
|
|
ss=i[0].replace('\\','/')
|
|
ss=ss.split('/')[-1].split('.')
|
|
name=self.download_path+ss[0]+' -'+str(id)+'.'+ss[1]
|
|
with open(name,'+bw') as f:
|
|
f.write(i[1])
|
|
return name
|
|
print("can not find the file with id:",id)
|
|
return ""
|
|
|
|
# 获取md5
|
|
def get_md5(self,id:int):
|
|
cmd="select md5 from {d1} WHERE id={d2};".format(d1=self.table_name,d2=id)
|
|
self.cur.execute(cmd)
|
|
a=self.cur.fetchall()[0]
|
|
return a[0]
|
|
|
|
# 扫描文件
|
|
def scan_files(self):
|
|
path = self.download_path
|
|
if not os.path.exists(path):
|
|
os.makedirs(path)
|
|
list=os.listdir(path)
|
|
return list
|
|
# 判断文件是否存在
|
|
def exists(self,id:int):
|
|
for i in self.scan_files():
|
|
s=i.split('.')[-2].split('-')[-1]
|
|
if(int(s)==id):
|
|
with open(self.download_path+i,"rb") as f:
|
|
md=md5(f.read())
|
|
if(md==self.get_md5(id)):
|
|
return True,i
|
|
return False,""
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
if(len(sys.argv)>=3):
|
|
sql_path=sys.argv[1]
|
|
sql_file=sys.argv[2]
|
|
s=sql()
|
|
if(s.init(sql_path)==True):
|
|
q=quest.quest_text("请输入本次提交的描述信息")
|
|
ack,text=q.show()
|
|
if(ack==True):
|
|
s.insert(sql_file,text)
|
|
else:
|
|
print("user cancelled.")
|
|
|
|
|
|
|
|
|