-
Notifications
You must be signed in to change notification settings - Fork 1
/
main.py
264 lines (225 loc) · 9.39 KB
/
main.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
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
import requests
import json
import os
from dotenv import load_dotenv
import logging
import shutil
from datetime import datetime
# 加载环境变量
load_dotenv()
# 在文件开头设置日志级别
logging.basicConfig(level=logging.DEBUG)
# 飞书应用的 App ID 和 App Secret
APP_ID = os.getenv("FEISHU_APP_ID")
APP_SECRET = os.getenv("FEISHU_APP_SECRET")
GPT4_API_KEY = os.getenv("GPT4_API_KEY")
BITABLE_APP_TOKEN = os.getenv("BITABLE_APP_TOKEN")
TABLE_ID = os.getenv("TABLE_ID")
# 添加新的常量
NEW_ACCOUNT_FILE = "new_account.txt"
ARCHIVE_FOLDER = "archived_accounts"
def get_tenant_access_token():
url = "https://open.feishu.cn/open-apis/auth/v3/tenant_access_token/internal"
headers = {
"Content-Type": "application/json"
}
data = {
"app_id": APP_ID,
"app_secret": APP_SECRET
}
response = requests.post(url, headers=headers, data=json.dumps(data))
if response.status_code == 200:
return response.json().get("tenant_access_token")
else:
print(f"获取 token 失败: {response.text}")
return None
def parse_text_with_gpt4(text):
url = "https://api.openai-next.com/v1/chat/completions"
headers = {
"Authorization": f"Bearer {GPT4_API_KEY}",
"Content-Type": "application/json"
}
system_prompt = system_prompt = '''
你是一个专门解析文本的AI助手。请将给定的文本解析为JSON格式，严格遵循以下要求：
1. 输出格式必须是一个包含 "records" 数组的JSON对象。
2. 每个记录应包含 "fields" 对象，其中包含 "产品"、"登录说明" 和 "账号信息" 三个字段。
3. "产品" 和 "登录说明" 对于同一产品的所有账号应该相同。
4. "账号信息" 应该是单行文本，包含完整的账号信息。
5. 不同产品的账号信息应该分开成不同的记录。
示例输入：
Cursor:
谷歌登录 账号-密码-辅助邮箱
[email protected]@pmail.1s.fr
[email protected]@outlook.com
Bolt:
会员-账号（-辅助邮箱）
[email protected]@hotmail.com
V0:
邮箱验证码登录 账号-密码
Reweb:
账号-密码
示例输出：
{
"records": [
{
"fields": {
"产品": "Cursor",
"登录说明": "谷歌登录 账号-密码-辅助邮箱",
"账号信息": "[email protected]@pmail.1s.fr"
}
},
{
"fields": {
"产品": "Cursor",
"登录说明": "谷歌登录 账号-密码-辅助邮箱",
"账号信息": "[email protected]@outlook.com"
}
},
{
"fields": {
"产品": "Bolt",
"登录说明": "会员-账号（-辅助邮箱）",
"账号信息": "[email protected]@hotmail.com"
}
},
{
"fields": {
"产品": "V0",
"登录说明": "邮箱验证码登录 账号-密码",
"账号信息": "[email protected]"
}
},
{
"fields": {
"产品": "Reweb",
"登录说明": "账号-密码",
"账号信息": "[email protected]"
}
}
]
}
注意事项：
1. 严格遵循示例输出的JSON结构。
2. 确保每个账号信息都是一个单独的记录。
3. 不要添加任何额外的注释或说明，只返回纯JSON数据。
4. 确保JSON格式正确，可以被直接解析。
5. 如果输入中包含多个产品，请为每个产品的每个账号创建单独的记录。
6. 返回的结果只能是JSON数据，不能有其他非JSON字符，例如不要有 "```json" 或 "```"。
'''
data = {
"model": "gpt-4o",
"messages": [
{
"role": "system",
"content": system_prompt
},
{
"role": "user",
"content": f"{text}"
}
],
"temperature": 0.7
}
response = requests.post(url, headers=headers, json=data)
logging.debug(f"GPT-4 API 响应状态码: {response.status_code}")
logging.debug(f"GPT-4 API 响应头: {response.headers}")
if response.status_code == 200:
response_json = response.json()
logging.debug(f"GPT-4 API 完整响应: {response_json}")
parsed_content = response_json["choices"][0]["message"]["content"]
logging.info(f"GPT-4 返回的内容: {parsed_content}")
try:
parsed_data = json.loads(parsed_content)
# 确保"产品"字段是一个字符串
if "产品" in parsed_data and isinstance(parsed_data["产品"], list):
parsed_data["产品"] = parsed_data["产品"][0] if parsed_data["产品"] else ""
return parsed_data
except json.JSONDecodeError as e:
logging.error(f"JSON 解析错误: {e}")
logging.error(f"GPT-4 返回的内容不是有效的 JSON 格式: {parsed_content}")
return None
else:
logging.error(f"GPT-4 API 调用失败: {response.text}")
return None
def read_new_account_file():
if not os.path.exists(NEW_ACCOUNT_FILE):
logging.error(f"文件 {NEW_ACCOUNT_FILE} 不存在")
return None
with open(NEW_ACCOUNT_FILE, 'r', encoding='utf-8') as file:
content = file.read()
return content
def archive_file_content():
if not os.path.exists(ARCHIVE_FOLDER):
os.makedirs(ARCHIVE_FOLDER)
if os.path.exists(NEW_ACCOUNT_FILE):
try:
with open(NEW_ACCOUNT_FILE, 'r', encoding='utf-8') as file:
content = file.read()
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
archive_filename = f"account_{timestamp}.txt"
archive_path = os.path.join(ARCHIVE_FOLDER, archive_filename)
with open(archive_path, 'w', encoding='utf-8') as archive_file:
archive_file.write(content)
logging.info(f"文件内容已归档: {archive_path}")
except IOError as e:
logging.error(f"归档文件内容时发生错误: {e}")
else:
logging.warning(f"文件 {NEW_ACCOUNT_FILE} 不存在，无法归档")
def clear_file_content():
if os.path.exists(NEW_ACCOUNT_FILE):
try:
archive_file_content() # 先归档文件内容
with open(NEW_ACCOUNT_FILE, 'w', encoding='utf-8') as file:
file.write('')
logging.info(f"文件 {NEW_ACCOUNT_FILE} 内容已清空")
except IOError as e:
logging.error(f"清空文件 {NEW_ACCOUNT_FILE} 时发生错误: {e}")
else:
logging.warning(f"文件 {NEW_ACCOUNT_FILE} 不存在，无需清空")
def add_records_to_bitable(token, parsed_data):
url = f"https://open.feishu.cn/open-apis/bitable/v1/apps/{BITABLE_APP_TOKEN}/tables/{TABLE_ID}/records/batch_create"
headers = {
"Authorization": f"Bearer {token}",
"Content-Type": "application/json"
}
if not parsed_data or "records" not in parsed_data or not parsed_data["records"]:
logging.error("没有有效的记录可以添加到 Bitable")
return False
data = {"records": parsed_data["records"]}
logging.info(f"正在向 Bitable 添加记录，URL: {url}")
logging.debug(f"请求头: {headers}")
logging.debug(f"请求数据: {data}")
response = requests.post(url, headers=headers, json=data)
logging.info(f"Bitable API 响应状态码: {response.status_code}")
if response.status_code == 200:
logging.info("成功添加记录到 Bitable")
logging.debug(f"Bitable API 响应: {response.json()}")
return True
else:
response_json = response.json()
logging.error(f"添加记录失败: {response_json}")
logging.error(f"BITABLE_APP_TOKEN: {BITABLE_APP_TOKEN}")
logging.error(f"TABLE_ID: {TABLE_ID}")
if response_json.get('code') == 1254062:
logging.error("单选字段转换失败，需要人工检查")
return False
def main():
token = get_tenant_access_token()
if token:
logging.info(f"成功获取 token: {token}")
content = read_new_account_file()
if content is None:
return
parsed_data = parse_text_with_gpt4(content)
if parsed_data:
success = add_records_to_bitable(token, parsed_data)
if success:
clear_file_content() # 这个函数现在会先归档内容，然后清空文件
else:
logging.error("添加记录失败，文件内容未清空和归档")
else:
logging.error("无法解析文本，跳过添加记录到 Bitable")
if __name__ == "__main__":
main()