mirror of
https://github.com/OpenBMB/ChatDev.git
synced 2024-11-07 18:40:13 +03:00
48 lines
1.9 KiB
Python
48 lines
1.9 KiB
Python
import re
|
|
import os
|
|
import time
|
|
from colorama import Fore
|
|
|
|
|
|
class Documents():
|
|
def __init__(self, generated_content = "", parse = True, predifined_filename = None):
|
|
self.directory: str = None
|
|
self.generated_content = generated_content
|
|
self.docbooks = {}
|
|
|
|
if generated_content != "":
|
|
if parse:
|
|
regex = r"```\n(.*?)```"
|
|
matches = re.finditer(regex, self.generated_content, re.DOTALL)
|
|
for match in matches:
|
|
filename = "requirements.txt"
|
|
doc = match.group(1)
|
|
self.docbooks[filename] = doc
|
|
else:
|
|
self.docbooks[predifined_filename] = self.generated_content
|
|
|
|
def _update_docs(self, generated_content, parse = True, predifined_filename = ""):
|
|
new_docs = Documents(generated_content, parse, predifined_filename)
|
|
for key in new_docs.docbooks.keys():
|
|
if key not in self.docbooks.keys() or self.docbooks[key] != new_docs.docbooks[key]:
|
|
print("{} updated.".format(key))
|
|
print(Fore.WHITE + "------Old:\n{}\n------New:\n{}".format(self.docbooks[key] if key in self.docbooks.keys() else "# None", new_docs.docbooks[key]))
|
|
self.docbooks[key] = new_docs.docbooks[key]
|
|
|
|
|
|
def _rewrite_docs(self):
|
|
directory = self.directory
|
|
if not os.path.exists(directory):
|
|
os.mkdir(directory)
|
|
print("{} Created.".format(directory))
|
|
for filename in self.docbooks.keys():
|
|
with open(os.path.join(directory, filename), "w", encoding="utf-8") as writer:
|
|
writer.write(self.docbooks[filename])
|
|
print(os.path.join(directory, filename), "Writen")
|
|
|
|
def _get_docs(self):
|
|
content = ""
|
|
for filename in self.docbooks.keys():
|
|
content += "{}\n```\n{}\n```\n\n".format(filename, self.docbooks[filename])
|
|
return content
|