-
Notifications
You must be signed in to change notification settings - Fork 0
/
gui.py
182 lines (163 loc) · 7.17 KB
/
gui.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
import requests
import tkinter as tk
import tkinter.messagebox
import tkinter.filedialog
import tkinter.simpledialog
import time
import json
import openai
import os
print('''
__ __ __ __
/ // /__ _/ /__ ___ ____ ___/ /
/ _ / _ `/ / _ \/ _ `/ -_) _ /
/_//_/\_,_/_/\___/\_, /\__/\_,_/ v1.2.2
/___/
''')
def clean_hitokoto():
os.system("del log.txt")
tkinter.messagebox.showinfo("提示","删除成功！")
def ope_help_doc():
os.system("start https://www.yuque.com/haloged/rwmrh5")
def zdy():
zdy_url=tkinter.simpledialog.askstring(title="请输入api地址",prompt = "地址")
zdy_j=tkinter.simpledialog.askstring(title="请输入一言文字所在的json路径",prompt="路径（直接输出输入zjscc）：")
zdy_cs=tkinter.simpledialog.askinteger(title="请输入运行次数",prompt="次数：")
if zdy_j=="zjscc":
with open("log.txt","a") as f:
f.write("获取源：自定义源"+"\n"+"api地址："+zdy_url+"\n抓取次数："+str(zdy_cs)+"\n\n")
for i in range(zdy_cs):
zdy_get=requests.get(zdy_url)
print(zdy_get.text())
with open("log.txt","a") as f:
f.write(zdy_get.text()+"\n")
time.sleep(2)
with open("log.txt","a") as f:
f.write("\n\n")
tkinter.messagebox.showinfo("提示","抓取成功！")
else:
with open("log.txt","a") as f:
f.write("获取源：自定义源"+"\napi地址："+zdy_url+"\n抓取次数："+str(zdy_cs)+"\n\n")
for i in range(zdy_cs):
zdy_jx={zdy_j:"1"}
zdy_get=requests.get(zdy_url)
zdy_jx_get=json.loads(zdy_get.text)
print(zdy_jx_get[zdy_j])
with open("log.txt","a") as f:
f.write(zdy_jx_get[zdy_j]+"\n")
time.sleep(2)
with open("log.txt","a") as f:
f.write("\n\n")
tkinter.messagebox.showinfo("提示","抓取成功！")
def ope_config():
tkinter.messagebox.showinfo("提示","正在开发中，预计1.5.0版本上线！")
def jcgx():
vertion=requests.get("https://tinywebdb.appinventor.space/api?user=haloged&secret=463de003&action=get&tag=bbh")
vertion_jx=json.loads(vertion.text)
bbh=vertion_jx["bbh"]
if bbh=="1.2.2":
tkinter.messagebox.showinfo("提示","无更新")
else:
tip_vertion=tkinter.messagebox.askyesno("提示","有新版本！\n点击“确定”转到仓库")
if tip_vertion==True:
os.system("start https://github.com/haloged/get_hitokoto/releases")
def ope_github():
os.system("start https://github.com/haloged/get_hitokoto/")
def ope():
os.system("log.txt")
def about():
tkinter.messagebox.showinfo("关于软件","作者：haloged\n软件版本：1.2.2\n作者B站：https://space.bilibili.com/518055250\nGithub仓库：https://github.com/haloged/get_hitokoto")
def run_1():
yuan=var.get()
run_num=tkinter.simpledialog.askinteger(title="请输入运行次数",prompt = "运行次数：")
if yuan==0:
with open("log.txt","a") as f:
f.write("获取源：hitokoto.cn\n抓取次数："+str(run_num)+"\n\n")
for i in range(run_num):
hitokoto_jx={"hitokoto":"1"}
hitokoto_get=requests.get("https://v1.hitokoto.cn")
hitokoto_jx=json.loads(hitokoto_get.text)
print(hitokoto_jx["hitokoto"])
with open("log.txt","a") as f:
f.write(hitokoto_jx["hitokoto"]+"\n")
time.sleep(2)
with open("log.txt","a") as f:
f.write("\n\n")
tkinter.messagebox.showinfo("提示","抓取成功！")
elif yuan==1:
with open("log.txt","a") as f:
f.write("获取源：apiup\n抓取次数："+str(run_num)+"\n\n")
for i in range(run_num):
apiup_jx={"txt":"1"}
apiup_get=requests.get("https://apiup.tcxone.eu.org/yiyan.php")
apiup_jx=json.loads(apiup_get.text)
print(apiup_jx["txt"])
with open("log.txt","a") as f:
f.write(apiup_jx["txt"]+"\n")
time.sleep(2)
with open("log.txt","a") as f:
f.write("\n\n")
tkinter.messagebox.showinfo("提示","获取成功！")
elif yuan==2:
with open("log.txt","a") as f:
f.write("获取源：韩小韩API\n抓取次数："+str(run_num)+"\n\n")
for i in range(run_num):
vvhan_get=requests.get("https://api.vvhan.com/api/ian")
print(vvhan_get.text)
with open("log.txt","a") as f:
f.write(vvhan_get.text+"\n")
time.sleep(2)
with open("log.txt","a") as f:
f.write("\n\n")
tkinter.messagebox.showinfo("提示","获取成功！")
elif yuan==3:
with open("log.txt","a") as f:
f.write("获取源：ChatGPT\n抓取次数："+str(run_num)+"\n\n")
gpt_api_token=tkinter.simpledialog.askstring(title="请输入你的你的KEY",prompt="请输入你的OPENAI API KEY")
openai.api_key = os.getenv(gpt_api_token)
response = openai.Completion.create(model="text-davinci-003", prompt="请生成"+run_num+"个关于励志的句子", temperature=0, max_tokens=1000)
print(response)
with open("log.txt","a") as f:
f.write(response)
with open("log.txt","a") as f:
f.write("\n\n")
tkinter.messagebox.showinfo("提示","获取成功！")
root=tk.Tk()
root.title("一言生成器v1.2.2 By Haloged")
root.geometry("500x300")
mainmenu = tk.Menu(root)
menuFile = tk.Menu(mainmenu) # 菜单分组 menuFile
mainmenu.add_cascade(label="文件",menu=menuFile)
menuFile.add_command(label="打开配置文件",command=ope_config)
menuFile.add_command(label="打开一言保存文件",command=ope)
menuFile.add_command(label="清空一言保存文件",command=clean_hitokoto)
menuFile.add_command(label="自定义源",command=zdy)
menuFile.add_separator() # 分割线
menuFile.add_command(label="退出",command=root.destroy)
menuEdit = tk.Menu(mainmenu) # 菜单分组 menuEdit
mainmenu.add_cascade(label="更多",menu=menuEdit)
menuEdit.add_command(label="帮助文档",command=ope_help_doc)
menuEdit.add_command(label="Github仓库",command=ope_github)
menuEdit.add_command(label="检查更新",command=jcgx)
menuEdit.add_command(label="关于",command=about)
root.config(menu=mainmenu)
logo=tk.Label(root,text="一言生成器",font=("宋体",20))
logo.pack()
br=tk.Label(root,text=" ",font=("宋体",20))
br.pack()
run=tk.Button(root,text="开始抓取",command=run_1)
run.pack()
br1=tk.Label(root,text=" ",font=("宋体",20))
br1.pack()
xz=tk.Label(root,text="抓取源")
xz.pack()
var = tk.IntVar()
rd1 = tk.Radiobutton(root,text="hitokoto.cn",variable=var,value=0)
rd1.pack()
rd2 = tk.Radiobutton(root,text="apiup",variable=var,value=1)
rd2.pack()
rd3 = tk.Radiobutton(root,text="韩小韩API",variable=var,value=2)
rd3.pack()
rd4 = tk.Radiobutton(root,text="ChatGPT(需自行提供API KEY)",variable=var,value=3)
rd4.pack()
root.mainloop()