RuwoScan/File.py at main · Tangruwo/RuwoScan · GitHub

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
# 就是唐,不想写注释了
import json
import sys
import os

def saveLog(path,diaName,Content): # 保存日志
    try:
        with open(path,"a",encoding="utf-8") as log:
            wMess = {
                "role": diaName, "content": Content
            }

            log.write(json.dumps(wMess, ensure_ascii=False) + "\n")
    except FileNotFoundError as e:
        print("文件不存在:", path, e)
        sys.exit(1)

def getLog(path): # 读取日志
    try:
        data = []

        with open(path,"r",encoding="utf-8") as log:
            for line in log:
                if line.strip():
                    entry = json.loads(line)
                    content = entry.get("content")

                    if content is None:
                        content = ""
                    elif isinstance(content, dict):
                        content = json.dumps(content, ensure_ascii=False)
                    elif not isinstance(content, str):
                        content = str(content)

                    data.append({"role": "user", "content": content})
        return data
    except FileNotFoundError as e:
        print("文件不存在:", path,e)
        return []

    except Exception as e:
        print("错误:",e)
        sys.exit(1)

def getConfig(): # 从config文件中获取配置文件
    try:
        with open("config.json","r+",encoding="utf-8") as f:
            if f.read() == '':
                apiKey = input(">> no apiKey,plase: ")
                bashUrl = input(">> no bashUrl,plase: ")
                model = input(">> no model,plase: ")

                config = {
                    "apiKey": apiKey,
                    "bashUrl": bashUrl,
                    "model": model
                }

                json.dump(config,f,indent=4)

                return config

            else:
                f.seek(0) # 指针归0 话说今天是八月十四了,www我的暑假

                config = json.load(f)
                return config

    except FileNotFoundError: # 呜呜太可恶了竟然乱删我的东西(17.14好了现在不上传啦)
        with open("config.json","w",encoding="utf-8") as f: # 复制还能水一点代码,酣畅淋漓
            apiKey = input(">>没有检测到apiKey,请输入: ")
            bashUrl = input(">>没有检测到bashUrl,请输入: ")
            model = input(">>没有检测到model,请输入: ")
            print("文件已储存到config.json,注意保密")

            config = {
                "apiKey": apiKey,
                "bashUrl": bashUrl,
                "model": model
            }
            json.dump(config,f,indent=4)
            f.seek(0)

            return config

    except Exception as e:
        print("错误:",e)
        sys.exit(1)


def initFile(): # 初始化文件
    filesPath = [
        "prompts/ruwoscan.log",
        "prompts/King.log",
        "prompts/battle.log"
    ]

    dirsPath = [
        "bugReport",
        "Downloads"
    ]

    try:
        # 文件初始化
        for file in filesPath:
            with open(file,"w",encoding="utf-8") as file:
                file.write("")

        # 文件夹初始化
        for dir in dirsPath:
            os.makedirs(dir, exist_ok=True)

    except Exception as e:
        print("初始化失败:", e)
        sys.exit(1)