fz0x1/.dotfiles
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

20250117.1737145183

Browse source
This commit is contained in:
fz0x1 2025-01-17 21:19:43 +01:00
parent df2469e105
commit ba81851219
1 changed files with 23 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
global/scripts/bin/diary.py
View file

@ -7,6 +7,7 @@ import logging
import os import os
import re import re
import shlex import shlex
import shutil
import sqlite3 import sqlite3
import subprocess import subprocess
import sys import sys
@ -273,6 +274,10 @@ def doctor():
diary_name = sys.argv[2] diary_name = sys.argv[2]
diary_path = get_diary_path_by_name(diary_name).parent 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) conn = db_connection(diary_path)
initialize_db(conn) initialize_db(conn)
@ -328,12 +333,25 @@ def make_hash(file: Path):
return sha256_hash.hexdigest() 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(): def export():
if len(sys.argv) < 4 or sys.argv[1] != "export": if len(sys.argv) < 4 or sys.argv[1] != "export":
sys.exit("Usage: script.py export <diary_name> <tag>") sys.exit("Usage: script.py export <diary_name> <tag>")
diary_name, tag = sys.argv[2], sys.argv[3] diary_name, tag = sys.argv[2], sys.argv[3]
diary_path = get_diary_path_by_name(diary_name).parent 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: try:
conn = db_connection(diary_path) conn = db_connection(diary_path)
@ -399,10 +417,15 @@ def insert():
# sys.exit("Invalid usage for bulk insert.") # sys.exit("Invalid usage for bulk insert.")
diary_name = sys.argv[2] diary_name = sys.argv[2]
# do backup ever!
insert_type = ( insert_type = (
"bulk" if len(sys.argv) > 3 and sys.argv[3] == "bulk" else "single" "bulk" if len(sys.argv) > 3 and sys.argv[3] == "bulk" else "single"
) )
diary_path = get_diary_path_by_name(diary_name).parent 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) conn = db_connection(diary_path)
initialize_db(conn) initialize_db(conn)