Skip to content

Commit d25270a

Browse files
committed
fix: replace print statements with logging in repo_load_controller.py
Signed-off-by: ANJAN672 <rockyanjan672@gmail.com>
1 parent 99aacb8 commit d25270a

File tree

1 file changed

+14
-15
lines changed

1 file changed

+14
-15
lines changed

augur/util/repo_load_controller.py

Lines changed: 14 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def add_cli_repo(self, repo_data: Dict[str, Any], from_org_list=False, repo_type
6161
try:
6262
repo_type = result[1]["repo_type"]
6363
except KeyError:
64-
print("Skipping repo type...")
64+
logger.debug("Skipping repo type...")
6565

6666

6767
# if the repo doesn't exist it adds it
@@ -94,59 +94,58 @@ def add_cli_org(self, org_name):
9494
repos = result[0]
9595
type = result[1]["owner_type"]
9696
if not repos:
97-
print(
98-
f"No organization with name {org_name} could be found")
97+
logger.warning(f"No organization with name {org_name} could be found")
9998
return {"status": "No organization found"}
10099

101100
# check if the repo group already exists
102101
query = self.session.query(RepoGroup).filter(RepoGroup.rg_name == org_name)
103102
rg = execute_session_query(query, 'first')
104103
if rg:
105-
print(f"{rg.rg_name} is already a repo group")
104+
logger.info(f"{rg.rg_name} is already a repo group")
106105

107106
return {"status": "Already a repo group"}
108107

109-
print(f'Organization "{org_name}" found')
108+
logger.info(f'Organization "{org_name}" found')
110109

111110
rg = RepoGroup(rg_name=org_name, rg_description="", rg_website="", rg_recache=0, rg_type="Unknown",
112111
tool_source="Loaded by user", tool_version="1.0", data_source="Git")
113112
self.session.add(rg)
114113
self.session.commit()
115114
repo_group_id = rg.repo_group_id
116-
print(f"{org_name} repo group created")
115+
logger.info(f"{org_name} repo group created")
117116

118117
for repo_url in repos:
119-
print(f"Adding {repo_url}")
118+
logger.info(f"Adding {repo_url}")
120119
result, status = self.add_cli_repo({"url": repo_url, "repo_group_id": repo_group_id}, from_org_list=True, repo_type=type)
121120
if not result:
122-
print(status["status"])
121+
logger.warning(status["status"])
123122

124123
return {"status": "Org added"}
125124

126125

127126
def paginate_repos(self, source, page=0, page_size=25, sort="repo_id", direction="ASC", **kwargs):
128127

129128
if not source:
130-
print("Func: paginate_repos. Error: Source Required")
129+
logger.error("Func: paginate_repos. Error: Source Required")
131130
return None, {"status": "Source Required"}
132131

133132
if source not in ["all", "user", "group"]:
134-
print("Func: paginate_repos. Error: Invalid source")
133+
logger.error("Func: paginate_repos. Error: Invalid source")
135134
return None, {"Invalid source"}
136135

137136
if direction and direction != "ASC" and direction != "DESC":
138-
print("Func: paginate_repos. Error: Invalid direction")
137+
logger.error("Func: paginate_repos. Error: Invalid direction")
139138
return None, {"status": "Invalid direction"}
140139

141140
try:
142141
page = int(page) if page else 0
143142
page_size = int(page_size) if page else 25
144143
except TypeError:
145-
print("Func: paginate_repos. Error: Page size and page should be integers")
144+
logger.error("Func: paginate_repos. Error: Page size and page should be integers")
146145
return None, {"status": "Page size and page should be integers"}
147146

148147
if page < 0 or page_size < 0:
149-
print("Func: paginate_repos. Error: Page size and page should be positive")
148+
logger.error("Func: paginate_repos. Error: Page size and page should be positive")
150149
return None, {"status": "Page size and page should be postive"}
151150

152151
order_by = sort if sort else "repo_id"
@@ -183,11 +182,11 @@ def paginate_repos(self, source, page=0, page_size=25, sort="repo_id", direction
183182
def get_repo_count(self, source, **kwargs):
184183

185184
if not source:
186-
print("Func: get_repo_count. Error: Source Required")
185+
logger.error("Func: get_repo_count. Error: Source Required")
187186
return None, {"status": "Source Required"}
188187

189188
if source not in ["all", "user", "group"]:
190-
print("Func: get_repo_count. Error: Invalid source")
189+
logger.error("Func: get_repo_count. Error: Invalid source")
191190
return None, {"status": "Invalid source"}
192191

193192
query, result = self.generate_repo_query(source, count=True, **kwargs)

0 commit comments

Comments
 (0)