forked from karimhabush/cyberowl
-
Notifications
You must be signed in to change notification settings - Fork 1
/
mdtemplate.py
40 lines (34 loc) · 1.47 KB
/
mdtemplate.py
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
# Returns the template of a bulletin
class Template:
def __init__(self, _source, _data):
self.SOURCE = _source
self.DATA = _data
# Data validation
self._validate_data()
def _validate_data(self):
for row in self.DATA:
if "_title" not in row:
raise ValueError(
"The dictionnaries in _data arrow is expecting the element _title.")
if "_desc" not in row:
raise ValueError(
"The dictionnaries in _data arrow is expecting the element _desc.")
if "_link" not in row:
raise ValueError(
"The dictionnaries in _data arrow is expecting the element _link.")
if "_date" not in row:
raise ValueError(
"The dictionnaries in _data arrow is expecting the element _date.")
def _set_heading(self):
return f"""## {self.SOURCE}\n"""
def _set_table_headers(self):
return """|Title|Description|Date|\n|---|---|---|\n"""
def _set_table_content(self, _title, _link, _desc, _date):
return f"""| [{_title}]({_link}) | {_desc} | {_date} |\n"""
def _fill_table(self):
TABLE = self._set_heading()
TABLE += self._set_table_headers()
for row in self.DATA:
TABLE += self._set_table_content(row["_title"],
row["_link"], row["_desc"], row["_date"])
return TABLE