pilot.linkstec 0.0.32__py3-none-any.whl → 0.0.91__py3-none-any.whl

pilot/db/create_table.py

@@ -0,0 +1,34 @@
+from pathlib import Path
+
+from db.sql_executor import exec_ddl
+
+
+this_path = Path(__file__).resolve()
+
+this_dir = this_path.parent
+
+
+
+
+
+ddl_file_path_list = [
+    # this_dir / 'ddl' / 'METHOD_CLASS_INFO.sql',
+    # this_dir / 'ddl' / 'METHOD_FILE_INFO.sql',
+    this_dir / 'ddl' / 'METHOD_INFO.sql',
+    # this_dir / 'ddl' / 'METHOD_PARAMETER_INFO.sql',
+    # this_dir / 'ddl' / 'METHOD_PARSING_INFO.sql',
+    # this_dir / 'ddl' / 'METHOD_SUB_METHOD_INFO.sql',
+    # this_dir / 'ddl' / 'SQL_CONDITION_INFO.sql',
+    # this_dir / 'ddl' / 'SQL_CONTENT_INFO.sql',
+    # this_dir / 'ddl' / 'SQL_FILE_INFO.sql',
+    # this_dir / 'ddl' / 'SQL_TABLE_INFO.sql',
+    # this_dir / 'ddl' / 'SQL_PARAMETER_INFO.sql',
+    # this_dir / 'ddl' / 'SQL_PARSING_INFO.sql',
+    # this_dir / 'ddl' / 'SQL_COLUMN_INFO.sql',
+    # this_dir / 'ddl' / 'TABLE_INFO.sql',
+    # this_dir / 'ddl' / 'TABLE_COLUMN_INFO.sql'
+]
+
+
+for ddl_file_path in ddl_file_path_list:
+    exec_ddl(str(ddl_file_path))

pilot/db/db_connect.py

@@ -0,0 +1,49 @@
+import sqlite3
+from pathlib import Path
+from contextlib import contextmanager
+from typing import Generator
+
+# -------------------- 配置 --------------------
+DB_FILE = Path(__file__).with_name("app.sqlite3")   # 默认数据库文件
+# ------------------------------------------------
+
+def _create_connection(db_path: Path = DB_FILE) -> sqlite3.Connection:
+    """
+    创建并返回一个 SQLite 连接对象。
+    - 自动创建文件（如果不存在）。
+    - 开启 WAL 模式，提高并发读取性能。
+    """
+    conn = sqlite3.connect(db_path, detect_types=sqlite3.PARSE_DECLTYPES)
+    # 让查询结果可以像字典一样访问（可选）
+    conn.row_factory = sqlite3.Row
+    # 开启 Write‑Ahead Logging（一次性设置即可）
+    conn.execute("PRAGMA journal_mode=WAL;")
+    return conn
+
+
+@contextmanager
+def get_connection(db_path: Path = DB_FILE) -> Generator[sqlite3.Connection, None, None]:
+    """
+    上下文管理器：自动打开连接、提交事务并在结束时关闭。
+    使用方式：
+        with get_connection() as conn:
+            conn.execute(...)
+    """
+    conn = _create_connection(db_path)
+    try:
+        yield conn
+        conn.commit()          # 正常退出时提交事务
+    except Exception:         # 发生异常则回滚
+        conn.rollback()
+        raise
+    finally:
+        conn.close()
+
+
+def get_raw_connection(db_path: Path = DB_FILE) -> sqlite3.Connection:
+    """
+    如果你不想使用上下文管理器，而是自行控制生命周期，
+    可以直接调用此函数获取一个 Connection 对象。
+    注意：使用完后务必手动 `conn.close()`
+    """
+    return _create_connection(db_path)

pilot/db/db_main.py

@@ -0,0 +1,293 @@
+from pathlib import Path
+
+from db.sql_executor import exec_ddl, exec_dml, exec_multiple_dml
+from db.sql_service import select_sql_info
+
+
+# メソッド別のファイル情報を登録する
+def insert_file_info(params):
+    # パッケージ名
+    package_name = params.get('package_name')
+    # ファイルパス
+    file_path = params.get('file_path')
+    # ファイル名
+    file_name = params.get('file_name')
+    # コメントなしのファイルパス
+    no_comment_file_path = params.get('no_comment_file_path')
+    # パラメーター情報をマッピング情報に設定する
+    dml_params = {
+        "package_name": package_name,
+        "file_name": file_name,
+        "original_file_path": file_path,
+        "no_comment_file_path": no_comment_file_path,
+        "comment_file_path": "XXXX"
+    }
+    #
+    dml_file_path = r'ddl\INSERT_METHOD_FILE_INFO.sql'
+
+    exec_dml(dml_file_path, dml_params)
+
+
+def insert_class_info(params):
+
+    package_name = params.get('package_name')
+
+    file_name = params.get('file_name')
+
+    class_name = params.get('class_name')
+
+    class_type = params.get('class_type')
+
+    full_class_name = params.get('full_class_name')
+
+    import_object_name = params.get('import_object_name')
+
+    dml_params = {
+        "package_name": package_name,
+        "file_name": file_name,
+        "class_name": class_name,
+        "class_type": class_type,
+        "full_class_name": full_class_name,
+        "import_object_name": import_object_name
+
+    }
+
+    dml_file_path = r'ddl\INSERT_METHOD_CLASS_INFO.sql'
+
+    exec_dml(dml_file_path, dml_params)
+
+
+def insert_method_info(params):
+
+    package_name = params.get('package_name')
+
+    file_name = params.get('file_name')
+
+    class_name = params.get('class_name')
+
+    class_type = params.get('class_type')
+
+    full_class_name = params.get('full_class_name')
+
+    import_object_name = params.get('import_object_name')
+
+    dml_params = {
+        "package_name": package_name,
+        "file_name": file_name,
+        "class_name": class_name,
+        "class_type": class_type,
+        "full_class_name": full_class_name,
+        "import_object_name": import_object_name
+
+    }
+
+    dml_file_path = r'ddl\INSERT_METHOD_INFO.sql'
+
+    exec_dml(dml_file_path, dml_params)
+
+
+def insert_method_parameter_info(params):
+
+    package_name = params.get('package_name')
+
+    file_name = params.get('file_name')
+
+    class_name = params.get('class_name')
+
+    class_type = params.get('class_type')
+
+    full_class_name = params.get('full_class_name')
+
+    import_object_name = params.get('import_object_name')
+
+    dml_params = {
+        "package_name": package_name,
+        "file_name": file_name,
+        "class_name": class_name,
+        "class_type": class_type,
+        "full_class_name": full_class_name,
+        "import_object_name": import_object_name
+
+    }
+
+    dml_file_path = r'ddl\INSERT_METHOD_PARAMETER_INFO.sql'
+
+    exec_dml(dml_file_path, dml_params)
+
+
+def insert_sub_method_info(params):
+
+    package_name = params.get('package_name')
+
+    file_name = params.get('file_name')
+
+    class_name = params.get('class_name')
+
+    class_type = params.get('class_type')
+
+    full_class_name = params.get('full_class_name')
+
+    import_object_name = params.get('import_object_name')
+
+    dml_params = {
+        "package_name": package_name,
+        "file_name": file_name,
+        "class_name": class_name,
+        "class_type": class_type,
+        "full_class_name": full_class_name,
+        "import_object_name": import_object_name
+
+    }
+
+    dml_file_path = r'ddl\INSERT_SUB_METHOD_INFO.sql'
+
+    exec_dml(dml_file_path, dml_params)
+
+
+def insert_sql_file_info(params):
+
+    mapper_file_name = params.get('mapper_file_name')
+
+    original_file_path = params.get('original_file_path')
+
+    no_comment_file_path = params.get('no_comment_file_path')
+
+    dml_params = {
+        "mapper_file_name": mapper_file_name,
+        "original_file_path": original_file_path,
+        "no_comment_file_path": no_comment_file_path
+    }
+
+    dml_file_path = r'ddl\INSERT_SQL_FILE_INFO.sql'
+
+    exec_dml(dml_file_path, dml_params)
+
+
+def insert_sql_info(params):
+
+    sql_id = params.get('sql_id')
+
+    sql_style = params.get('sql_style')
+
+    mapper_file_name = params.get('mapper_file_name')
+
+    table_name = params.get('table_name')
+
+    table_alias = params.get('table_alias')
+
+    dml_params = {
+        "sql_id": sql_id,
+        "sql_style": sql_style,
+        "mapper_file_name": mapper_file_name,
+        "table_name" : table_name,
+        "table_alias" : table_alias
+    }
+
+    dml_file_path = r'ddl\INSERT_SQL_INFO.sql'
+
+    exec_dml(dml_file_path, dml_params)
+
+
+def insert_condition_info(params):
+    sql_id = params.get('sql_id')
+
+    sql_style = params.get('sql_style')
+
+    mapper_file_name = params.get('mapper_file_name')
+
+    condition_content = params.get('condition_content')
+
+    dml_params = {
+        "sql_id": sql_id,
+        "sql_style": sql_style,
+        "mapper_file_name": mapper_file_name,
+        "condition_content" : condition_content
+    }
+
+    dml_file_path = r'ddl\INSERT_CONDITION_INFO.sql'
+
+    exec_dml(dml_file_path, dml_params)
+
+def insert_parameter_info(params):
+    sql_id = params.get('sql_id')
+
+    sql_style = params.get('sql_style')
+
+    mapper_file_name = params.get('mapper_file_name')
+
+    condition_content = params.get('condition_content')
+
+    dml_params = {
+        "sql_id": sql_id,
+        "sql_style": sql_style,
+        "mapper_file_name": mapper_file_name,
+        "condition_content" : condition_content
+    }
+
+    dml_file_path = r'ddl\INSERT_CONDITION_INFO.sql'
+
+    exec_dml(dml_file_path, dml_params)
+
+
+def insert_column_info(params):
+    sql_id = params.get('sql_id')
+
+    sql_style = params.get('sql_style')
+
+    mapper_file_name = params.get('mapper_file_name')
+
+    condition_content = params.get('condition_content')
+
+    dml_params = {
+        "sql_id": sql_id,
+        "sql_style": sql_style,
+        "mapper_file_name": mapper_file_name,
+        "condition_content" : condition_content
+    }
+
+    dml_file_path = r'ddl\INSERT_CONDITION_INFO.sql'
+
+    exec_dml(dml_file_path, dml_params)
+
+
+
+#
+#
+#
+
+if __name__ == "__main__":
+    # ddl_file_path = r'E:\METHOD_FILE_INFO.sql'
+    # exec_ddl(ddl_file_path)
+    #
+    # params = {
+    #     "package_name": "com.example.service",
+    #     "file_name": "OrderService.py",
+    #     "original_file_path": "/src/com/example/service/OrderService.py",
+    #     "no_comment_file_path": "/tmp/no_comment/OrderService.py",
+    #     "comment_file_path": "/tmp/commented/OrderService.py"
+    # }
+
+    # search_sql_info = select_sql_info('sql_info', {'sql_id': 'batchInsertAddress'})
+
+
+    # dml_file_path = r'E:\SELECT_SQL_PARAMETER_INFO.sql'
+    #
+    # dml_file_path = r'E:\SELECT_SQL_TABLE_INFO.sql'
+    #
+    # dml_file_path = r'E:\SELECT_METHOD_SUB_METHOD_INFO.sql'
+    #
+    dml_file_path = r'E:\SELECT_METHOD_INFO.sql'
+    result = exec_dml(dml_file_path)
+
+    for row_info in result:
+
+        print(dict(row_info))
+
+    # #
+    # dml_file_path = r'E:\DROP_TABLE_METHOD.sql'
+    # dml_file_path = r'E:\DELETE_METHOD_INFO.sql'
+    # # dml_file_path = r'E:\UPDATE_METHOD_SUB_METHOD_INFO.sql'
+    # result = exec_multiple_dml(dml_file_path)
+
+
+