fixed eid missed

data/2.PostgreSQL中阶/1.PostgreSQL数据库的基本结构/6.数据库扩展/extension.json

@@ -2,5 +2,6 @@
   "type": "code_options",
   "author": "ccat",
   "source": "extension.md",
-  "notebook_enable": false
+  "notebook_enable": false,
+  "exercise_id": "fad124e7fe1a420dbfbe72de8d4b3f11"
 }
\ No newline at end of file

data/2.PostgreSQL中阶/1.PostgreSQL数据库的基本结构/6.数据库扩展/language.json

@@ -2,5 +2,6 @@
   "type": "code_options",
   "author": "ccat",
   "source": "language.md",
-  "notebook_enable": false
+  "notebook_enable": false,
+  "exercise_id": "4657a4c34eec421a913625fb61af00d4"
 }
\ No newline at end of file

data/2.PostgreSQL中阶/2.服务端编程/1.聚合和分组/salary.json

@@ -2,5 +2,6 @@
   "type": "code_options",
   "author": "ccat",
   "source": "salary.md",
-  "notebook_enable": false
+  "notebook_enable": false,
+  "exercise_id": "31deab40b4af44c38c363b2333bdd977"
 }
\ No newline at end of file

data/2.PostgreSQL中阶/2.服务端编程/2.连接查询/customer_order.json

@@ -2,5 +2,6 @@
   "type": "code_options",
   "author": "ccat",
   "source": "customer_order.md",
-  "notebook_enable": false
+  "notebook_enable": false,
+  "exercise_id": "18d5a22c34874bb4b73aac6846674243"
 }
\ No newline at end of file

data/2.PostgreSQL中阶/2.服务端编程/5.CTE/to_root.json

@@ -2,5 +2,6 @@
   "type": "code_options",
   "author": "ccat",
   "source": "to_root.md",
-  "notebook_enable": false
+  "notebook_enable": false,
+  "exercise_id": "45f0cef12fcf4f9e90276dff6a336657"
 }
\ No newline at end of file

data/3.PostgreSQL高阶/1.集群和复制/3.fdw/fdw.json

@@ -2,5 +2,6 @@
   "type": "code_options",
   "author": "ccat",
   "source": "fdw.md",
-  "notebook_enable": false
+  "notebook_enable": false,
+  "exercise_id": "762a9c3f032b4ce884858c18e2bb580c"
 }
\ No newline at end of file

requirements.txt

 pre_commit~=2.16.0
+GitPython~=3.1.24
\ No newline at end of file

src/tree.py

@@ -4,6 +4,8 @@ import os
 import re
 import sys
 import uuid
+import re
+import git
 
 id_set = set()
 logger = logging.getLogger(__name__)
@@ -12,7 +14,10 @@ handler = logging.StreamHandler(sys.stdout)
 formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
 handler.setFormatter(formatter)
 logger.addHandler(handler)
+repo = git.Repo(".")
 
+def user_name():
+    return repo.config_reader().get_value("user", "name")
 
 def load_json(p):
     with open(p, 'r') as f:
@@ -301,6 +306,14 @@ class TreeWalker:
 
     def ensure_exercises(self, section_path):
         config = self.ensure_section_config(section_path)
+        for e in os.listdir(section_path):
+            base, ext = os.path.splitext(e)
+            _, source = os.path.split(e)
+            if ext != ".md":
+                continue
+            meta_path = os.path.join(section_path, base + ".json")
+            self.ensure_exercises_meta(meta_path, source)
+
         for e in config.get("export", []):
             full_name = os.path.join(section_path, e)
             exercise = load_json(full_name)
@@ -311,6 +324,36 @@ class TreeWalker:
             else:
                 id_set.add(exercise["exercise_id"])
 
+    def ensure_exercises_meta(self, meta_path, source):
+        _, mfile = os.path.split(meta_path)
+        if os.path.exists(meta_path):
+            meta = load_json(meta_path)
+            if "exercise_id" not in meta:
+                meta["exercise_id"] = uuid.uuid4().hex
+            if "notebook_enable" not in meta:
+                meta["notebook_enable"] = self.default_notebook()
+            if "source" not in meta:
+                meta["source"] = source
+            if "author" not in meta:
+                meta["author"] = user_name()
+            if "type" not in meta:
+                meta["type"] = "code_options"
+        else:
+            meta = {
+                "type": "code_options",
+                "author": user_name(),
+                "source": source,
+                "notebook_enable": self.default_notebook(),
+                "exercise_id": uuid.uuid4().hex
+            }
+        dump_json(meta_path, meta, True, True)
+
+    def default_notebook(self):
+        if self.name in ["python", "java", "c", "algorithm"]:
+            return True
+        else:
+            return False
+
     def check_section_keywords(self, full_path):
         config = self.ensure_section_config(full_path)
         if not config.get("keywords", []):