summaryrefslogtreecommitdiff
path: root/note_tools/__init__.py
blob: 69e6e67fd2add36771cacf159110116043878924 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
from datetime import datetime
import os
from typing import List, Optional, Tuple


class NoteBaseCommand(object):    
  date = None

  def file_name(self):
    return f"{self.date.strftime('%Y%m%d')}.md"

  def get_formatted_dates(self) -> Tuple[str, str]:
    return (self.date.strftime("%Y%m%d"), self.date.strftime("%Y-%m-%d"))

  def get_date(self) -> datetime:
    return datetime.now()

  def get_date_parts(self) -> Tuple[str]:
    return (self.date.strftime('%Y'), self.date.strftime('%m'), self.date.strftime('%d'))
      
  def write_file_heading(self, file_path):      
    date_yyyymmdd = self.date.strftime('%Y-%m-%d') 
    underline = '=' * len(date_yyyymmdd)
    with open(file_path, 'w') as file:
      file.write(f"{date_yyyymmdd}\n{underline}\n\n\n")

  def parse_file_date (self, file_name: str) -> Tuple[str, str]:
    year = file_name[:4]
    month = file_name[4:6]
    return year, month

  def get_notes_path(self, project_data) -> Optional[str]:
    if project_data and 'folders' in project_data:
      for folder in project_data['folders']:
        folder_path = folder['path']
        if os.path.basename(folder_path).lower() == "notes":
          return folder_path
    return None