From 6edd63b9669087601fc510cbcebb7f54258ad4e0 Mon Sep 17 00:00:00 2001 From: "Karl O. Pinc" Date: Sun, 1 Oct 2023 14:59:29 -0500 Subject: [PATCH] Add db comments for columns --- db/schemas/gen_comments.py | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/db/schemas/gen_comments.py b/db/schemas/gen_comments.py index dbdda03..b3da1b4 100755 --- a/db/schemas/gen_comments.py +++ b/db/schemas/gen_comments.py @@ -148,8 +148,6 @@ class DBObj: sdir = typedir.parent if self.objtype == "functions": self.sql = self.function_lookup(sdir, objname) - elif self.objtype == "tables" and objname.find(".") != -1: - self.sql = [SQL["column"].format(name=objname)] else: self.sql = [SQL[self.objtype].format(name=objname)] @@ -221,14 +219,23 @@ def comments_from_file(file, db_objs): if line[0:indent] == prefix: comment.append(line[indent:]) else: - db_obj = db_objs[obj] - db_obj.cf.write(db_obj.sql, "".join(comment)[:-1]) + db_obj = db_objs[obj_name] + if column: + sql = [SQL["column"].format(name=obj)] + else: + sql = db_obj.sql + db_obj.cf.write(sql, "".join(comment)[:-1]) in_match = False if not in_match: match = OBJ_SUMMARY.match(line) if match is not None: obj = match.group(2).lower() - if obj in db_objs: + column = obj.find(".") != -1 + if column: + obj_name = obj.split(".")[0] + else: + obj_name = obj + if obj_name in db_objs: indent = len(match.group(1)) prefix = " " * indent comment = [match.group(3).lstrip()] -- 2.34.1