From f67b171385e5366fc33190285dde61b6d37feaa9 Mon Sep 17 00:00:00 2001
From: Soulter <905617992@qq.com>
Date: Sun, 19 May 2024 17:10:11 +0800
Subject: [PATCH] =?UTF-8?q?perf:=20=E6=95=B0=E6=8D=AE=E5=BA=93=E8=BF=81?=
 =?UTF-8?q?=E7=A7=BB=E8=87=B3=20data=20=E7=9B=AE=E5=BD=95=E4=B8=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 persist/session.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/persist/session.py b/persist/session.py
index c9cbb80ac..a0a45114e 100644
--- a/persist/session.py
+++ b/persist/session.py
@@ -1,13 +1,16 @@
 import sqlite3
-import yaml
+import os
+import shutil
 import time
 from typing import Tuple
 
 
 class dbConn():
     def __init__(self):
-        # 读取参数,并支持中文
-        conn = sqlite3.connect("data.db")
+        db_path = "data/data.db"
+        if os.path.exists("data.db"):
+            shutil.copy("data.db", db_path)
+        conn = sqlite3.connect(db_path)
         conn.text_factory = str
         self.conn = conn
         c = conn.cursor()