Skip to content
This repository has been archived by the owner on May 27, 2024. It is now read-only.

Commit

Permalink
Minor modifications
Browse files Browse the repository at this point in the history
  • Loading branch information
bastgau committed Mar 1, 2024
1 parent 3224520 commit 142fbf4
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 1 deletion.
4 changes: 4 additions & 0 deletions src/snow_revoke_privileges/snow_privileges.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,6 +56,8 @@ def prepare_future_false(self) -> None:

self.progress.next()

self.progress.finish()

def prepare_future_false_task(self, row: Any) -> Optional[pd.DataFrame]:
"""..."""

Expand Down Expand Up @@ -101,6 +103,8 @@ def prepare_future_true(self) -> None:
for _ in pool.imap_unordered(self.prepare_future_true_task, only_database_schema.iterrows()): # pyright: ignore
self.progress.next()

self.progress.finish()

def prepare_future_true_task(self, row: Any) -> Optional[pd.DataFrame]:
"""..."""

Expand Down
2 changes: 1 addition & 1 deletion src/snow_revoke_privileges/tools/my_snowflake.py
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,7 @@ def execute_multi_requests(requests: List[str]) -> None: # pylint: disable=unus
requests (List[str]): A list of SQL queries to be executed on a Snowflake database connection.
"""

with Bar("Executing request in Snwoflake", max=len(requests)) as progress:
with Bar("Executing request in Snowflake", max=len(requests)) as progress:

with Pool(processes=8) as pool:
for _ in pool.imap_unordered(MySnowflake.execute_single_request, requests): # pyright: ignore
Expand Down

0 comments on commit 142fbf4

Please sign in to comment.