From ff117453a9ac841aa3ad60678f113082239ed85f Mon Sep 17 00:00:00 2001
From: Ivan Reshetnikov <ordinarydev@protonmail.com>
Date: Wed, 24 Aug 2022 22:43:57 +0500
Subject: [PATCH] Simplify the process of creating dirs and files

---
 histd.py | 26 +++++++++-----------------
 1 file changed, 9 insertions(+), 17 deletions(-)

diff --git a/histd.py b/histd.py
index 8c83e49..aeecca1 100755
--- a/histd.py
+++ b/histd.py
@@ -17,15 +17,9 @@ def get_base_dir() -> str:
 
     Returns the path to the directory where data can be stored.
     """
-    local_dir = os.path.expanduser('~/.local')
-    if not os.path.exists(local_dir):
-        os.mkdir(local_dir)
-
-    histd_dir = os.path.join(local_dir, 'histd')
-    if not os.path.exists(histd_dir):
-        os.mkdir(histd_dir)
-
-    return histd_dir
+    base_dir = os.path.expanduser('~/.local/histd')
+    os.makedirs(base_dir, exist_ok=True)
+    return base_dir
 
 
 def edit_note(base_dir: str, note_date: date):
@@ -34,16 +28,14 @@ def edit_note(base_dir: str, note_date: date):
     so that the user can describe the day.
     """
     # Create dirs (base_dir/year/month)
-    year_dir = os.path.join(base_dir, str(note_date.year))
-    if not os.path.exists(year_dir):
-        os.mkdir(year_dir)
-
-    month_dir = os.path.join(year_dir, f'{note_date.month:02}')
-    if not os.path.exists(month_dir):
-        os.mkdir(month_dir)
+    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)
 
     # Open file (base_dir/year/month/day.txt) with default editor
-    path_to_file = os.path.join(month_dir, f'{note_date.day:02}.txt')
+    filename = f'{note_date.day:02}.txt'
+    path_to_file = os.path.join(workdir, filename)
     editor = os.environ.get('EDITOR', 'nano')
     subprocess.run([editor, path_to_file])