rgfewfger

models/__init__.py

@@ -1,7 +1,7 @@
-from peewee import *
-
-# 连接到 SQLite 数据库，如果文件不存在会自动创建
-db = SqliteDatabase(r'E:\新建文件夹\lm_job\models\database.db')
+# from peewee import *
+#
+# # 连接到 SQLite 数据库，如果文件不存在会自动创建
+# db = SqliteDatabase(r'E:\新建文件夹\lm_job\models\database.db')
 
 import pymysql
 
@@ -29,29 +29,29 @@ db1 = MySQLDatabase(
 )
 
 
-class BaseModel(Model):
-    class Meta:
-        database = db1
-
-    def save(self, *args, **kwargs):
-        """在调用 save 时自动连接和关闭（若无事务）"""
-        db.connect(reuse_if_open=True)
-        try:
-            result = super().save(*args, **kwargs)
-        finally:
-            # 若当前没有事务且连接仍然打开，则关闭连接
-            if not db.in_transaction() and not db.is_closed():
-                db.close()
-        return result
-
-    @classmethod
-    def get_or_create(cls, defaults=None, **kwargs):
-        """在调用 get_or_create 时自动连接和关闭（若无事务）"""
-        db.connect(reuse_if_open=True)
-        try:
-            obj, created = super().get_or_create(defaults=defaults, **kwargs)
-        finally:
-            # 若当前没有事务且连接仍然打开，则关闭连接
-            if not db.in_transaction() and not db.is_closed():
-                db.close()
-        return obj, created
+# class BaseModel(Model):
+#     class Meta:
+#         database = db1
+#
+#     def save(self, *args, **kwargs):
+#         """在调用 save 时自动连接和关闭（若无事务）"""
+#         db.connect(reuse_if_open=True)
+#         try:
+#             result = super().save(*args, **kwargs)
+#         finally:
+#             # 若当前没有事务且连接仍然打开，则关闭连接
+#             if not db.in_transaction() and not db.is_closed():
+#                 db.close()
+#         return result
+#
+#     @classmethod
+#     def get_or_create(cls, defaults=None, **kwargs):
+#         """在调用 get_or_create 时自动连接和关闭（若无事务）"""
+#         db.connect(reuse_if_open=True)
+#         try:
+#             obj, created = super().get_or_create(defaults=defaults, **kwargs)
+#         finally:
+#             # 若当前没有事务且连接仍然打开，则关闭连接
+#             if not db.in_transaction() and not db.is_closed():
+#                 db.close()
+#         return obj, created

models/ips.py

@@ -1,9 +1,9 @@
 from peewee import *
 
-from models import BaseModel, db1
+from models import  db1
 
 
-class Ips(BaseModel):
+class Ips(Model):
     id = IntegerField(primary_key=True)
     host = CharField(null=True)
     port = CharField(null=True)

models/xstart.py

@@ -1,9 +1,9 @@
 from peewee import *
 
-from models import db1, BaseModel
+from models import db1
 
 
-class Xstart(BaseModel):
+class Xstart(Model):
     id = AutoField(primary_key=True)  # 自增主键
     bit_id = IntegerField(null=True)
     start = CharField(null=True)

models/xtoken.py

@@ -1,9 +1,9 @@
 from peewee import *
 # 假设 db 已经在其他地方定义并连接到数据库
-from models import BaseModel, db1
+from models import  db1
 
 
-class XToken(BaseModel):
+class XToken(Model):
     id = AutoField(primary_key=True)  # 自增主键
     hub_id = IntegerField(null=True)  # hub_id 字段，整型，可为空
     start = IntegerField(null=True)  # start 字段，整型，可为空
@@ -19,5 +19,3 @@ class XToken(BaseModel):
         database = db1 # 所属数据库
         table_name = 'x_token'  # 表名
 
-# if __name__ == '__main__':
-#     XToken.create_table()