wangweiye
/
mathdeptv2
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

继续丰富database tools

This commit is contained in:
weiye.wang 2023-06-23 09:29:25 +08:00
parent 1277bfdbd4
commit 595b941397
1 changed files with 21 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
工具/database_tools.py
View File

@ -53,7 +53,8 @@ def generate_problem_series(startingid,length,adict): #在adict字典里返回
return ",".join(excludelist) #返回按顺序的题号列表
def generate_number_set(string): #根据可能含有":"和","的题号字符串生成一个用逗号分隔的六位题号列表, 例如"1:3,5"会生成["000001","000002","000003","000005"]
def generate_number_set(string,*thedict): #根据可能含有":"和","的题号字符串生成一个用逗号分隔的六位题号列表, 例如"1:3,5"会生成["000001","000002","000003","000005"]
#可变参数*dict如果存在, 将只生成dict的keys中包含的题号列表
string = re.sub(r"[\n\s]","",string)
string_list = string.split(",")
numbers_list = []
@ -64,7 +65,13 @@ def generate_number_set(string): #根据可能含有":"和","的题号字符串
start,end = s.split(":")
for ind in range(int(start),int(end)+1):
numbers_list.append(str(ind).zfill(6))
return numbers_list #返回六位题号列表
if len(thedict) == 0:
return numbers_list #返回六位题号列表
elif len(thedict) == 1 and type(thedict[0]) == dict:
numbers_list = [id for id in numbers_list if id in thedict[0]]
return numbers_list #返回字典中存在的六位题号列表
else:
return "输入参数有误"
def generate_exp(id_list): #根据题号列表生成字符串式的含":"和","的题号字符串, 例如["000001","000002","000003","000005"]生成"000001:000003,000005", 若列表为空则生成"无有效题号"
if not len(id_list) == 0:
@ -184,5 +191,17 @@ def CreateNewProblem(id,content,origin,dict,editor): # 构建一道新题目的
NewProblem["edit"] = [editor]
return NewProblem # 返回一道新题目的字典, 已赋新的ID, 内容, 来源和编辑者
def CreateIDLinks(old_id_list,new_id_list,*thedict): #建立已有id和新id之间的联系, thedict为可选, 选中的话即为当前字典, 会从new_id_list中排除当前字典中有的项
if len(thedict) == 1 and type(thedict[0]) == dict:
new_id_list = [id for id in new_id_list if not id in thedict[0]]
if len(old_id_list)>len(new_id_list):
return "新ID个数不足."
else:
id_links = []
for i in range(len(old_id_list)):
id_links.append((old_id_list[i],new_id_list[i]))
return id_links # 返回id联系, 每个元组表示一对id, 前者是旧id, 后者是新id
if __name__ == "__main__":
print("数据库工具, import用.")