@@ -15,66 +15,6 @@ def __init__(self, message, exit_code=None):
1515 super ().__init__ (self .message )
1616
1717
18- def setup (current_file ):
19- # Set up logging
20- logging .basicConfig (
21- level = logging .INFO ,
22- format = "%(asctime)s - %(levelname)s - %(module)s - %(message)s" ,
23- )
24- logger = logging .getLogger (__name__ )
25-
26- # Datetime
27- datetime_today = datetime .now (timezone .utc )
28- quarter = PeriodIndex ([datetime_today .date ()], freq = "Q" )[0 ]
29-
30- # Paths
31- paths = {}
32- paths ["repo" ] = os .path .dirname (path_join (__file__ , ".." ))
33- paths ["dotenv" ] = path_join (paths ["repo" ], ".env" )
34- paths ["data" ] = os .path .dirname (
35- os .path .abspath (os .path .realpath (current_file ))
36- )
37- current_phase = os .path .basename (
38- os .path .dirname (os .path .abspath (os .path .realpath (current_file )))
39- )
40- paths ["data" ] = path_join (paths ["repo" ], "data" )
41- data_quarter = path_join (paths ["data" ], f"{ quarter } " )
42- for phase in ["1-fetch" , "2-process" , "3-report" ]:
43- paths [f"data_{ phase } " ] = path_join (data_quarter , phase )
44- paths ["data_phase" ] = path_join (data_quarter , current_phase )
45-
46- paths ["data_quarter" ] = data_quarter
47-
48- return logger , paths
49-
50-
51- def paths_update (logger , paths , old_quarter , new_quarter ):
52- logger .info (f"Updating paths: replacing { old_quarter } with { new_quarter } " )
53- for label in [
54- "data_1-fetch" ,
55- "data_2-process" ,
56- "data_3-report" ,
57- "data_phase" ,
58- "data_quarter" ,
59- ]:
60- paths [label ] = paths [label ].replace (old_quarter , new_quarter )
61- return paths
62-
63-
64- def paths_log (logger , paths ):
65- paths_list = []
66- repo_path = paths ["repo" ]
67- for label , path in paths .items ():
68- label = f"{ label } :"
69- if label == "repo:" :
70- paths_list .append (f"\n { ' ' * 4 } { label } { path } " )
71- else :
72- path_new = path .replace (repo_path , "." )
73- paths_list .append (f"\n { ' ' * 8 } { label :<15} { path_new } " )
74- paths_list = "" .join (paths_list )
75- logger .info (f"PATHS:{ paths_list } " )
76-
77-
7818def git_fetch_and_merge (args , repo_path , branch = None ):
7919 if not args .enable_git :
8020 return
@@ -146,6 +86,66 @@ def path_join(*paths):
14686 return os .path .abspath (os .path .realpath (os .path .join (* paths )))
14787
14888
89+ def paths_log (logger , paths ):
90+ paths_list = []
91+ repo_path = paths ["repo" ]
92+ for label , path in paths .items ():
93+ label = f"{ label } :"
94+ if label == "repo:" :
95+ paths_list .append (f"\n { ' ' * 4 } { label } { path } " )
96+ else :
97+ path_new = path .replace (repo_path , "." )
98+ paths_list .append (f"\n { ' ' * 8 } { label :<15} { path_new } " )
99+ paths_list = "" .join (paths_list )
100+ logger .info (f"PATHS:{ paths_list } " )
101+
102+
103+ def paths_update (logger , paths , old_quarter , new_quarter ):
104+ logger .info (f"Updating paths: replacing { old_quarter } with { new_quarter } " )
105+ for label in [
106+ "data_1-fetch" ,
107+ "data_2-process" ,
108+ "data_3-report" ,
109+ "data_phase" ,
110+ "data_quarter" ,
111+ ]:
112+ paths [label ] = paths [label ].replace (old_quarter , new_quarter )
113+ return paths
114+
115+
116+ def setup (current_file ):
117+ # Set up logging
118+ logging .basicConfig (
119+ level = logging .INFO ,
120+ format = "%(asctime)s - %(levelname)s - %(module)s - %(message)s" ,
121+ )
122+ logger = logging .getLogger (__name__ )
123+
124+ # Datetime
125+ datetime_today = datetime .now (timezone .utc )
126+ quarter = PeriodIndex ([datetime_today .date ()], freq = "Q" )[0 ]
127+
128+ # Paths
129+ paths = {}
130+ paths ["repo" ] = os .path .dirname (path_join (__file__ , ".." ))
131+ paths ["dotenv" ] = path_join (paths ["repo" ], ".env" )
132+ paths ["data" ] = os .path .dirname (
133+ os .path .abspath (os .path .realpath (current_file ))
134+ )
135+ current_phase = os .path .basename (
136+ os .path .dirname (os .path .abspath (os .path .realpath (current_file )))
137+ )
138+ paths ["data" ] = path_join (paths ["repo" ], "data" )
139+ data_quarter = path_join (paths ["data" ], f"{ quarter } " )
140+ for phase in ["1-fetch" , "2-process" , "3-report" ]:
141+ paths [f"data_{ phase } " ] = path_join (data_quarter , phase )
142+ paths ["data_phase" ] = path_join (data_quarter , current_phase )
143+
144+ paths ["data_quarter" ] = data_quarter
145+
146+ return logger , paths
147+
148+
149149def update_readme (
150150 args ,
151151 section_title ,
0 commit comments