Skip to content

Commit 5ed2a7d

Browse files
committed
sort functions
1 parent c7a763d commit 5ed2a7d

File tree

1 file changed

+60
-60
lines changed

1 file changed

+60
-60
lines changed

scripts/shared.py

Lines changed: 60 additions & 60 deletions
Original file line numberDiff line numberDiff line change
@@ -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-
7818
def 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+
149149
def update_readme(
150150
args,
151151
section_title,

0 commit comments

Comments
 (0)