yorb_symlinker.py

# This software is under a Yet-To-Be-Licensed (YTBL) meta-license.
# Contribution is encouraged, but code is not yet ready to be released.
# THIS CODE SHALL NOT BE USED FOR ANY TYPE OF GENERATIVE MODEL TRAINING.
# That said, please feel free to contribute to the main project ("yorb-save"), until a license is finalized.
# Written by Collyn Townley, (ytbl) 2023
from os import path, symlink, curdir, remove, makedirs, rmdir
from shutil import move
from psutil import disk_usage
import yorb_indexer
import datetime


def create_archive():
    vault_path = path.abspath(curdir)
    if int(disk_usage(vault_path).percent) > 98:
        print("Insufficent space in current location to continue. Exiting.")
        exit(1)
    else:
        archive_name = input("Give your archive a name, or just press ENTER:")
        if archive_name == "":
            archive_name = (
                datetime.datetime.now().isoformat(timespec="minutes").replace(":", "-")
            )
        origin_path = input("Please enter the origin path:")
        dest_path = path.join(vault_path, "archive", archive_name, "")
        dest_file = dest_path + path.basename(origin_path)
        dest_file_basename = path.basename(origin_path)
        path_exists = path.exists(dest_path)
        if path_exists is False:
            makedirs(dest_path)
        move(origin_path, dest_file)
        symlink(dest_file, origin_path)
        current_date_time = datetime.datetime.now()
        archive_db_record = (
            archive_name,
            origin_path,
            dest_file,
            current_date_time,
        )
        yorb_indexer.add_archive_record(archive_db_record)
        print("Done!\a")
        print(f"Symlink created at: {origin_path}")


def delete_archive():
    vault_path = path.abspath(curdir)
    if int(disk_usage(vault_path).percent) > 98:
        print("Insufficent space in current location to continue. Exiting.")
        exit(1)
    else:
        if yorb_indexer.fetch_all_archives() == False:
            return
        else:
            archive_db_record = input("Select Archive to remove:")  # TODO: handle input
            archive_db_record_entry = yorb_indexer.fetch_selected_archive_record(
                archive_db_record
            )
            vault_path = archive_db_record_entry[2]
            origin_path = archive_db_record_entry[1]
            remove(origin_path)
            move(vault_path, origin_path)
            rmdir(path.dirname(vault_path))
            print(f"File restored to: {origin_path}")
            yorb_indexer.delete_archive_record(archive_db_record)


def debug_create_dummy_archive_record():
    archive_name = "null"
    dest_file_basename = "null"
    dest_file = "null"
    current_date_time = datetime.datetime.now()
    archive_db_record = (archive_name, dest_file_basename, dest_file, current_date_time)
    yorb_indexer.add_archive_record(archive_db_record)