20250117.1737145183
This commit is contained in:
parent
df2469e105
commit
ba81851219
1 changed files with 23 additions and 0 deletions
|
@ -7,6 +7,7 @@ import logging
|
|||
import os
|
||||
import re
|
||||
import shlex
|
||||
import shutil
|
||||
import sqlite3
|
||||
import subprocess
|
||||
import sys
|
||||
|
@ -273,6 +274,10 @@ def doctor():
|
|||
|
||||
diary_name = sys.argv[2]
|
||||
diary_path = get_diary_path_by_name(diary_name).parent
|
||||
diary_filename = get_diary_path_by_name(diary_name)
|
||||
|
||||
# do backup ever!
|
||||
make_backup(diary_filename, diary_path)
|
||||
|
||||
conn = db_connection(diary_path)
|
||||
initialize_db(conn)
|
||||
|
@ -328,12 +333,25 @@ def make_hash(file: Path):
|
|||
return sha256_hash.hexdigest()
|
||||
|
||||
|
||||
def make_backup(diary_name: str, diary_path: Path):
|
||||
try:
|
||||
os.remove(diary_path / f"{diary_name}.bak")
|
||||
except FileNotFoundError:
|
||||
...
|
||||
# shutil.copy2(diary_path / diary_name, diary_path / f"{diary_name}.bak") # I can't make backup when the diary is not encrypted...
|
||||
shutil.copy2(diary_path / DB_NAME, diary_path / f"{DB_NAME}.bak")
|
||||
|
||||
|
||||
def export():
|
||||
if len(sys.argv) < 4 or sys.argv[1] != "export":
|
||||
sys.exit("Usage: script.py export <diary_name> <tag>")
|
||||
|
||||
diary_name, tag = sys.argv[2], sys.argv[3]
|
||||
diary_path = get_diary_path_by_name(diary_name).parent
|
||||
diary_filename = get_diary_path_by_name(diary_name)
|
||||
|
||||
# do backup ever!
|
||||
make_backup(diary_filename, diary_path)
|
||||
|
||||
try:
|
||||
conn = db_connection(diary_path)
|
||||
|
@ -399,10 +417,15 @@ def insert():
|
|||
# sys.exit("Invalid usage for bulk insert.")
|
||||
|
||||
diary_name = sys.argv[2]
|
||||
# do backup ever!
|
||||
insert_type = (
|
||||
"bulk" if len(sys.argv) > 3 and sys.argv[3] == "bulk" else "single"
|
||||
)
|
||||
diary_path = get_diary_path_by_name(diary_name).parent
|
||||
diary_filename = get_diary_path_by_name(diary_name)
|
||||
|
||||
# do backup ever!
|
||||
make_backup(diary_filename, diary_path)
|
||||
|
||||
conn = db_connection(diary_path)
|
||||
initialize_db(conn)
|
||||
|
|
Loading…
Add table
Reference in a new issue