server-notes/notes.py at master · davidriegel/server-notes · GitHub

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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
import sys
import json
import time
import os

class printColors:
    HEADER = '\033[95m'
    OKBLUE = '\033[94m'
    OKCYAN = '\033[96m'
    OKGREEN = '\033[92m'
    WARNING = '\033[93m'
    FAIL = '\033[91m'
    ENDC = '\033[0m'
    BOLD = '\033[1m'
    UNDERLINE = '\033[4m'

def now() -> str:
    return time.strftime("%d/%m/%Y %H:%M:%S", time.localtime())

def createNote(title: str, note: str):
    json_data = open(os.path.join(os.path.dirname(__file__), 'notes.json'))
    data = json.load(json_data)
    json_data.close()

    for existing_note in data['notes']:
        if existing_note['title'] == title:
            return print(f"{printColors.FAIL}Note already exists{printColors.ENDC}")

    data['notes'].append({
        'createdAt': now(),
        'updatedAt': now(),
        'title': title,
        'note': note
    })

    json_data = open(os.path.join(os.path.dirname(__file__), 'notes.json'), 'w')
    json.dump(data, json_data)
    json_data.close()

    return print(f"{printColors.OKGREEN}Note created{printColors.ENDC}")

def removeNote(title: str):
    json_data = open(os.path.join(os.path.dirname(__file__), 'notes.json'))
    data = json.load(json_data)
    json_data.close()

    for existing_note in data['notes']:
        if existing_note['title'] == title:
            data['notes'].remove(existing_note)
            json_data = open(os.path.join(os.path.dirname(__file__), 'notes.json'), 'w')
            json.dump(data, json_data)
            json_data.close()

            return print(f"{printColors.OKGREEN}Note removed{printColors.ENDC}")

    return print(f"{printColors.FAIL}Note not found{printColors.ENDC}")


def showNote(title: str):
    if title == '':
        return showAllNotes()

    json_data = open(os.path.join(os.path.dirname(__file__), 'notes.json'))
    data = json.load(json_data)
    json_data.close()

    for note in data['notes']:
        if note['title'] == title:
            return print(f"{printColors.BOLD}{printColors.HEADER}Showing note: {printColors.UNDERLINE}{title}{printColors.ENDC}\n{printColors.OKCYAN}Created at: {note['createdAt']}{printColors.ENDC}\n{printColors.WARNING}Last update: {note['updatedAt']}{printColors.ENDC}\n\n{printColors.OKGREEN}{note['note']}{printColors.ENDC}")

    return print(f"{printColors.FAIL}Note not found{printColors.ENDC}")


def showAllNotes():
    json_data = open(os.path.join(os.path.dirname(__file__), 'notes.json'))
    data = json.load(json_data)
    json_data.close()

    notes = f"{printColors.BOLD}{printColors.HEADER}Showing all notes:{printColors.ENDC}\n\n"
    if len(data['notes']) == 0:
        return print(f"{printColors.FAIL}No notes found{printColors.ENDC}")

    for note in data['notes']:
        notes += f"{printColors.OKGREEN}Title:{printColors.ENDC} {printColors.UNDERLINE}{note['title']}{printColors.ENDC}\n"

    return print(notes)

def updateNote(title: str, note: str):
    json_data = open(os.path.join(os.path.dirname(__file__), 'notes.json'))
    data = json.load(json_data)
    json_data.close()

    for existing_note in data['notes']:
        if existing_note['title'] == title:
            existing_note['note'] = note
            existing_note['updatedAt'] = now()
            json_data = open(os.path.join(os.path.dirname(__file__), 'notes.json'), 'w')
            json.dump(data, json_data)
            json_data.close()

            return print(f"{printColors.OKGREEN}Note updated{printColors.ENDC}")

    return print(f"{printColors.FAIL}Note not found{printColors.ENDC}")

def showHelpMenu():
    return print(f"{printColors.HEADER}{printColors.UNDERLINE}{printColors.BOLD}This project is only here to take notes or quick thoughts from the CMD for later :){printColors.ENDC}\n{printColors.BOLD}I may add a UI at a later date{printColors.ENDC}\n\n{printColors.OKCYAN}py notes.py help {printColors.ENDC}| Shows you this menu.\n{printColors.OKCYAN}py add <title> <note> {printColors.ENDC}| Create a new note\n{printColors.OKCYAN}py remove <title> {printColors.ENDC}| Delete notes\n{printColors.OKCYAN}py update <title> <new note>{printColors.ENDC}| Update note\n{printColors.OKCYAN}py show {printColors.ENDC}| Overview of all notes\n{printColors.OKCYAN}py show <title> {printColors.ENDC}| Show note information{printColors.ENDC}")

if __name__ == '__main__':
    try:
        json_data = open(os.path.join(os.path.dirname(__file__), 'notes.json'))
        json_data.close()
    except FileNotFoundError:
        createJSON = open(os.path.join(os.path.dirname(__file__), 'notes.json'), 'w')
        createJSON.write('{"notes": []}')
        createJSON.close()

    if len(sys.argv) < 2:
        print('Usage: py notes.py <help|add|remove|show|update> <title> <optional:note>')
        sys.exit(1)

    action = sys.argv[1]
    title = sys.argv[2] if len(sys.argv) > 2 else ''
    note = sys.argv[3] if len(sys.argv) > 3 else ''

    if action in ["add", "update"] and len(sys.argv) < 4:
        print('Usage: py notes.py <add|update> <title> <note>')
        sys.exit(1)

    if len(title) > 25:
        print('Title must be less than 25 characters')
        sys.exit(1)

    if action == 'add':
        createNote(title=title, note=note)
    elif action == 'remove':
        removeNote(title=title)
    elif action == 'show':
        showNote(title=title)
    elif action == 'update':
        updateNote(title=title, note=note)
    elif action == 'help':
        showHelpMenu()
    else:
        print('Unknown action: ' + action)
        sys.exit(1)

    sys.exit(0)