wangzhibo
2025-07-27 a6662268312eb645cc7b5bf9f2ad6f5635d3cb28
auto_media_publisher/utils/utils_database.py
@@ -3,10 +3,12 @@
from collections import defaultdict
from datetime import datetime
from auto_media_publisher.config.conf_base import  DATABASE_PATH
from auto_media_publisher.utils.logger import PLATFORM_LOGGER
from auto_media_publisher.utils.logger import get_logger
DATABASE_FLAG_FILE = DATABASE_PATH.replace(".sqlite3",".initialized")
database_logger = get_logger("database","admin")
def _init_database():
    if os.path.exists(DATABASE_FLAG_FILE):
@@ -62,7 +64,7 @@
        # 创建标志文件
    with open(DATABASE_FLAG_FILE, 'w') as f:
        f.write("Database initialized")
        PLATFORM_LOGGER.info("Database initialized!!!")
        database_logger.info("Database initialized!!!")
def dajiala_insert_or_update_top15_json(json_obj):
    now = datetime.now().isoformat()
@@ -128,8 +130,7 @@
            ORDER BY last_update ASC
            LIMIT ?
        ''', (limit,))
        rows = cursor.fetchall()
        return rows
        return cursor.fetchall()
def dajiala_update_download_status(url, status):
    with sqlite3.connect(DATABASE_PATH) as conn:
@@ -169,7 +170,7 @@
            conn.commit()
        return True
    except Exception as e:
        PLATFORM_LOGGER.error("Database error!!! {e}")
        database_logger.error("Database error!!! {e}")
        return False
    
def publish_task_exists(user_id, platform, scheduled_time):
@@ -192,7 +193,7 @@
            conn.commit()
        return True
    except Exception as e:
        PLATFORM_LOGGER.error("Database error!!! {e}")
        database_logger.error("Database error!!! {e}")
        return False
def publish_task_delete(task_id):
@@ -202,7 +203,7 @@
            conn.commit()
        return True
    except Exception as e:
        PLATFORM_LOGGER.error("Database error!!! {e}")
        database_logger.error("Database error!!! {e}")
        return False
        
def platform_operation_fail_insert(user,platform,proxy,type_,note):
@@ -215,7 +216,7 @@
            conn.commit()
        return True
    except Exception as e:
        PLATFORM_LOGGER.error("Database error!!! {e}")
        database_logger.error("Database error!!! {e}")
        return False
def get_platform_upload_tasks(limit=200):
@@ -256,5 +257,5 @@
            conn.commit()
        return True
    except Exception as e:
        PLATFORM_LOGGER.error("Database error!!! {e}")
        database_logger.error("Database error!!! {e}")
        return False