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
|
import os
import sys
import logging
from logging import Logger
log: Logger = logging.getLogger(__name__)
# db class that works for both html and md files
class Database:
COLUMN_NUM: int = 4
def __init__(self, db_path: str):
log.debug('initializing the page db on path "%s"', db_path)
self.db_path: str = db_path
self.e: dict[str, tuple[float, float, list[str]]] = dict()
# updates the tags for a specific entry (file)
# file_name only contains the entry name (without the absolute path)
def update_tags(self, file_name: str,
tags: list[str]) -> None:
if file_name in self.e:
log.debug('updating tags for entry "%s"', file_name)
cts, mts, old_tags = self.e[file_name]
log.debug('entry "%s" old content: (%s, %s, (%s))',
file_name, cts, mts, ', '.join(old_tags))
self.e[file_name] = (cts, mts, tags)
log.debug('entry "%s" new content: (%s, %s, (%s))',
file_name, cts, mts, ', '.join(tags))
else:
log.error('can\'t update tags for entry "%s",'
' as it is not present in db', file_name)
sys.exit(1)
# returns a bool that indicates if the entry
# was (includes new entries) or wasn't updated
def update(self, file_name: str,
remove: str=None) -> bool:
log.debug('updating entry for file "%s"', file_name)
# initial default values
f: str = file_name
tags: list[str] = []
if remove is not None:
f = file_name.replace(remove, '')
log.debug('removed "%s" from "%s": "%s"', remove, file_name, f)
# get current time, needs actual file name
time: float = os.stat(file_name).st_mtime
log.debug('modified time for "%s": %f', file_name, time)
# three cases, 1) entry didn't exist,
# 2) entry hasn't been mod and,
# 3) entry has been mod
#1)
if f not in self.e:
log.debug('entry "%s" didn\'t exist, adding with defaults', f)
self.e[f] = (time, 0.0, tags)
return True
old_time, old_mod_time, tags = self.e[f]
log.debug('entry "%s" old content: (%s, %s, (%s))',
f, old_time, old_mod_time, ', '.join(tags))
# 2)
if old_mod_time == 0.0:
if time > old_time:
log.debug('entry "%s" has been modified for the first'
' time, updating', f)
self.e[f] = (old_time, time, tags)
log.debug('entry "%s" new content: (%s, %s, (%s))',
f, old_time, time, ', '.join(tags))
return True
# 3)
else:
if time > old_mod_time:
log.debug('entry "%s" has been modified, updating', f)
self.e[f] = (old_time, time, tags)
log.debug('entry "%s" new content: (%s, %s, (%s))',
f, old_time, time, ', '.join(tags))
return True
log.warning('none of the case scenarios for updating'
' entry "%s" matched', f)
return False
def write(self) -> None:
log.debug('writing db')
with open(self.db_path, 'w', newline='\n') as file:
# write each k,v pair in dict to db file
for k, v in self.e.items():
log.debug('parsing row for page "%s"', k)
t: str = None
row: str = None
if len(v[2]) == 0:
t = '-'
else:
t = ','.join(v[2])
row = f'{k} {v[0]} {v[1]} {t}'
log.debug('writing row: "%s"', row)
file.write(row)
def read(self) -> None:
if os.path.exists(self.db_path) and os.path.isfile(self.db_path):
rows: list[str] = None
log.debug('reading db')
with open(self.db_path, 'r') as file:
rows = file.readlines()
log.info('db contains %d rows', len(rows))
# parse each entry and populate accordingly
l: list[str] = None
# l=list of values in entry
log.debug('parsing rows from db')
for i, row in enumerate(rows):
log.debug('row %d content: "%s"', i, row)
l = tuple(row.strip().split())
if len(l) != self.COLUMN_NUM:
log.critical('row doesn\t contain %s columns,'
' contains %d elements; row %d content: "%s"',
self.COLUMN_NUM, len(l), i, row)
sys.exit(1)
t: list[str] = None
if l[3] == '-':
t = []
else:
t = l[3].split(',')
log.debug('tag content: (%s)', ', '.join(t))
self.e[l[0]] = (float(l[1]), float(l[2]), t)
else:
log.error('database file "%s" doesn\'t exist or it\'s not a file')
sys.exit(1)
|
