try to add database interface.

server/www/teleport/app/eom_app/app/core.py

@@ -18,6 +18,7 @@ import eom_common.eomcore.utils as utils
 from eom_common.eomcore.logger import log
 from .const import *
 from .configs import app_cfg
+from .db import db
 from .session import web_session
 
 cfg = app_cfg()
@@ -71,6 +72,8 @@ class WebServerCore:
         # db_path = os.path.join(cfg.data_path, 'ts_db.db')
         get_sqlite_pool().init(cfg.data_path)
 
+        db.init_sqlite(os.path.join(cfg.data_path, 'ts_db.db'))
+
         return True
 
     def _get_core_server_config(self):

server/www/teleport/app/eom_app/app/db.py

@@ -0,0 +1,100 @@
+# -*- coding: utf-8 -*-
+
+import os
+import threading
+import sqlite3
+
+from .configs import app_cfg
+from eom_common.eomcore.logger import log
+
+cfg = app_cfg()
+
+
+__all__ = ['db']
+
+
+class TPDatabase:
+    def __init__(self):
+        self.need_create = False  # 数据尚未存在，需要创建
+        self.need_upgrade = False  # 数据库已存在但版本较低，需要升级
+        self._conn_pool = None
+
+    def init_mysql(self):
+        # NOT SUPPORTED YET
+        pass
+
+    def init_sqlite(self, db_file):
+        self._conn_pool = TPSqlitePool(db_file)
+
+        if not os.path.exists(db_file):
+            self.need_create = True
+            return
+
+        # 看看数据库中是否存在用户表（如果不存在，可能是一个空数据库文件），则可能是一个新安装的系统
+        ret = self._conn_pool.query('SELECT COUNT(*) FROM sqlite_master where type="table" and name="ts_account";')
+        if ret[0][0] == 0:
+            self.need_create = True
+            return
+
+        # 尝试从配置表中读取当前数据库版本号（如果不存在，说明是比较旧的版本了，则置为0）
+        ret = self._conn_pool.query('SELECT value FROM ts_config where name="db_ver";')
+        if 0 == len(ret):
+            self.need_upgrade = True
+
+
+class TPDatabasePool:
+    def __init__(self):
+        self._locker = threading.RLock()
+        self._connections = dict()
+
+    def query(self, sql):
+        _conn = self._get_connect()
+        if _conn is None:
+            return list()
+        return self._do_query(_conn, sql)
+
+    def _get_connect(self):
+        with self._locker:
+            thread_id = threading.get_ident()
+            if thread_id not in self._connections:
+                _conn = self._do_connect()
+                self._connections[thread_id] = _conn
+            else:
+                _conn = self._connections[thread_id]
+
+        return _conn
+
+    def _do_connect(self):
+        return None
+
+    def _do_query(self, conn, sql):
+        return list()
+
+
+class TPSqlitePool(TPDatabasePool):
+    def __init__(self, db_file):
+        super().__init__()
+        self._db_file = db_file
+
+    def _do_connect(self):
+        try:
+            return sqlite3.connect(self._db_file)
+        except:
+            log.e('[sqlite] can not connect, does the database file correct?')
+            return None
+
+    def _do_query(self, conn, sql):
+        cursor = conn.cursor()
+        try:
+            cursor.execute(sql)
+            db_ret = cursor.fetchall()
+            return db_ret
+        except:
+            log.e('')
+            return list()
+        finally:
+            cursor.close()
+
+
+db = TPDatabase()
+del TPDatabase