cswimr/flake
cswimr/flake
Template
1
1
Fork 0
Code Issues 4 Pull requests Activity

script changes

Browse source
This commit is contained in:
Seaswimmer 2024-11-22 10:35:19 -05:00
parent 1745e834af
commit eaa112e0a8
Signed by: cswimr
GPG key ID: 0EC431A8DA8F8087
2 changed files with 11 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
scripts/common/common.py
View file

@ -30,8 +30,14 @@ def notify(
def read_secret_file(secret: str) -> str: def read_secret_file(secret: str) -> str:
path = f"/run/secrets/{secret}"
if not os.path.exists(path):
raise FileNotFoundError(f"Secret file {path} does not exist.")
with open(f"/run/secrets/{secret}", "r") as f: with open(f"/run/secrets/{secret}", "r") as f:
return f.read().strip() secret = f.read().strip()
if not secret:
raise ValueError(f"Secret file {path} is empty.")
return secret
def does_desktop_entry_exist(desktop_entry: str) -> bool: def does_desktop_entry_exist(desktop_entry: str) -> bool:

14
scripts/zipline.py
View file

@ -20,23 +20,17 @@ def zipline(
desktop_entry: str = None, desktop_entry: str = None,
) -> Any: ) -> Any:
token = read_secret_file("zipline") token = read_secret_file("zipline")
if not token:
raise FileNotFoundError(
"Secret file at /run/secrets/zipline either does not exist or is empty."
)
if not os.path.isfile(file_path): if not os.path.isfile(file_path):
raise FileNotFoundError(f"File at {file_path} does not exist.") raise FileNotFoundError(f"File at {file_path} does not exist.")
use_send_notify = False use_notify_send = False
if application_name and desktop_entry: if application_name and desktop_entry:
if not does_desktop_entry_exist(desktop_entry=desktop_entry): if not does_desktop_entry_exist(desktop_entry=desktop_entry):
raise FileNotFoundError("Desktop entry does not exist.") raise FileNotFoundError("Desktop entry does not exist.")
if not which("notify-send"): if not which("notify-send"):
raise FileNotFoundError("notify-send is not installed.") raise FileNotFoundError("notify-send is not installed.")
use_notify_send = True
use_send_notify = True
content_type = mimetypes.guess_type(file_path)[0] or "application/octet-stream" content_type = mimetypes.guess_type(file_path)[0] or "application/octet-stream"
@ -57,7 +51,7 @@ def zipline(
copy(text=link) copy(text=link)
print(f"Link copied to clipboard: {link}") print(f"Link copied to clipboard: {link}")
if use_send_notify: if use_notify_send:
notify( notify(
application_name=application_name, application_name=application_name,
title="Upload Successful", title="Upload Successful",
@ -73,7 +67,7 @@ def zipline(
raise Exception(error_message) raise Exception(error_message)
except Exception as e: except Exception as e:
if use_send_notify: if use_notify_send:
notify( notify(
application_name=application_name, application_name=application_name,
title="Upload Failed", title="Upload Failed",