Prechádzať zdrojové kódy

add gen_md_file_dirs translate get_now_time into utils

weichen 3 rokov pred
rodič
commit
ddf1608830
1 zmenil súbory, kde vykonal 22 pridanie a 0 odobranie
  1. 22 0
      models/utils/__init__.py

+ 22 - 0
models/utils/__init__.py

@@ -1,3 +1,8 @@
+from os import path, walk
+from googletrans import Translator
+from datetime import datetime, timezone, timedelta
+
+
 def write_md(f_dir, content):
     with open(f_dir, 'w') as md:
         md.write(content)
@@ -7,3 +12,20 @@ def read_line_md(f_dir):
     with open(f_dir, 'r') as md:
         pre_content = md.readlines()
     return pre_content
+
+
+def gen_md_file_dirs(dir_):
+    for root, dirs, files in walk(dir_):
+        for f in files:
+            if '.md' not in f:
+                continue
+            yield path.join(root, f)
+
+
+def translate(text):
+    result = Translator().translate(text, dest='en')
+    return result.text
+
+
+def get_now_time():
+    return datetime.now(timezone(timedelta(hours=+8))).isoformat(timespec="seconds")