| | |
| | | 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): |
| | |
| | | # 创建标志文件 |
| | | 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() |
| | |
| | | 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: |
| | |
| | | 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): |
| | |
| | | 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): |
| | |
| | | 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): |
| | |
| | | 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): |
| | |
| | | conn.commit() |
| | | return True |
| | | except Exception as e: |
| | | PLATFORM_LOGGER.error("Database error!!! {e}") |
| | | database_logger.error("Database error!!! {e}") |
| | | return False |