Browse Source

构建mangodb_cache数据库及连接代码,构建test测试代码

Air 4 weeks ago
parent
commit
e8f8e912c8
3 changed files with 34 additions and 16 deletions
  1. 0 16
      main.py
  2. 25 0
      src/kg_construction/mongodb_cache.py
  3. 9 0
      test/mangodb_test.py

+ 0 - 16
main.py

@@ -1,16 +0,0 @@
-# 这是一个示例 Python 脚本。
-
-# 按 Shift+F10 执行或将其替换为您的代码。
-# 按 双击 Shift 在所有地方搜索类、文件、工具窗口、操作和设置。
-
-
-def print_hi(name):
-    # 在下面的代码行中使用断点来调试脚本。
-    print(f'Hi, {name}')  # 按 Ctrl+F8 切换断点。
-
-
-# 按间距中的绿色按钮以运行脚本。
-if __name__ == '__main__':
-    print_hi('PyCharm')
-
-# 访问 https://www.jetbrains.com/help/pycharm/ 获取 PyCharm 帮助

+ 25 - 0
src/kg_construction/mongodb_cache.py

@@ -0,0 +1,25 @@
+import pymongo
+
+
+class MongoDBConn:
+    def initConnect(self, host_port, db_name, assemble):
+        """
+        建立数据库连接
+        """
+        # client = pymongo.MongoClient(host=host, port=port, username=user, password=password)
+        # db = client[db_name]
+        # collection = db[assemble]
+
+        myclient = pymongo.MongoClient(host_port)
+        mydb = myclient[db_name]
+        collection = mydb[assemble]
+
+        return collection
+
+if __name__ == "__main__":
+    host_port = 'mongodb://222.24.30.227:27017/'
+    db_name = 'nebula-kg-cache'
+    assemble = 'mechatronics-seed'
+    mongoDBConn = MongoDBConn()
+    collection = mongoDBConn.initConnect(host_port=host_port, db_name=db_name, assemble=assemble)
+    print(collection)

+ 9 - 0
test/mangodb_test.py

@@ -0,0 +1,9 @@
+from src.kg_construction.mongodb_cache import MongoDBConn
+
+if __name__ == "__main__":
+    host_port = 'mongodb://222.24.30.227:27017/'
+    db_name = 'nebula-kg-cache'
+    assemble = 'mechatronics-seed'
+    mongoDBConn = MongoDBConn()
+    collection = mongoDBConn.initConnect(host_port=host_port, db_name=db_name, assemble=assemble)
+    print(collection)