histd/histd.py

51 lines
1.4 KiB
Python
Raw Normal View History

2022-08-19 17:31:51 +05:00
#!/usr/bin/env python3
from datetime import date
import os
import subprocess
def main():
base_dir = get_base_dir()
today = date.today()
edit_note(base_dir, today)
def get_base_dir() -> str:
"""
Creates the directories necessary for the program
to work, if they are not present.
Returns the path to the directory where data can be stored.
"""
base_dir = os.path.expanduser('~/.local/share/histd')
os.makedirs(base_dir, exist_ok=True)
return base_dir
2022-08-19 17:31:51 +05:00
def edit_note(base_dir: str, note_date: date):
"""
Creates the required directories and opens a text editor
so that the user can describe the day.
"""
# Create dirs (base_dir/year/month)
year = str(note_date.year)
month = f'{note_date.month:02}'
workdir = os.path.join(base_dir, year, month)
os.makedirs(workdir, exist_ok=True)
2022-08-19 17:31:51 +05:00
2022-10-02 15:10:19 +05:00
# Open file (base_dir/year/month/day.md) with default editor
filename = f'{note_date.day:02}.md'
path_to_file = os.path.join(workdir, filename)
2022-08-19 17:31:51 +05:00
editor = os.environ.get('EDITOR', 'nano')
2022-09-30 13:34:44 +05:00
try:
subprocess.run([editor, path_to_file], check=True, cwd=base_dir)
except FileNotFoundError:
print("Error: I can't find your text editor")
print("Make sure the 'EDITOR' environment variable is set correctly")
except subprocess.CalledProcessError:
print("Your editor returned non-zero exit code")
2022-08-19 17:31:51 +05:00
if __name__ == '__main__':
main()