add: auto link feature

This commit is contained in:
h z
2025-01-17 16:33:39 +00:00
parent 00f6895850
commit 26b64f8c15
5 changed files with 49 additions and 23 deletions

View File

@@ -21,14 +21,14 @@ def get_db():
finally:
db.close()
def dump_db():
try:
os.environ['MYSQL_PWD'] = DB_PASSWORD
dump_cmd = f"mysqldump --no-tablespaces -h {DB_HOST} -P {DB_PORT} -u {DB_USER} {DB_NAME} > /app/dump/db_dump.sql"
subprocess.run(dump_cmd, shell=True, check=True)
except subprocess.CalledProcessError as e:
print(f"Failed to dump database: {e}")
raise e
# def dump_db():
# try:
# os.environ['MYSQL_PWD'] = DB_PASSWORD
# dump_cmd = f"mysqldump --no-tablespaces -h {DB_HOST} -P {DB_PORT} -u {DB_USER} {DB_NAME} > /app/dump/db_dump.sql"
# subprocess.run(dump_cmd, shell=True, check=True)
# except subprocess.CalledProcessError as e:
# print(f"Failed to dump database: {e}")
# raise e
def clear_db():
with engine.connect() as conn:
@@ -73,11 +73,11 @@ def init_payload():
def setup_db():
if DB_SCHEMA_UPDATED:
try:
dump_db()
print("[ x ] db dumped")
except Exception as e:
print(f"[ x ] Failed to dump database: {e}")
# try:
# dump_db()
# print("[ x ] db dumped")
# except Exception as e:
# print(f"[ x ] Failed to dump database: {e}")
clear_db()
print("[ x ] db cleared")
create_all()