Fix lint

Author: pvk-developer

github_analytics/__main__.py

@@ -190,7 +190,8 @@ def _get_parser():
 
     # Traffic
     traffic = action.add_parser(
-        'traffic', help='Collect github traffic metrics.', parents=[logging_args])
+        'traffic', help='Collect github traffic metrics.', parents=[logging_args]
+    )
     traffic.set_defaults(action=_traffic_collection)
 
     traffic.add_argument('-t', '--token', type=str, required=False, help='Github Token to use.')
@@ -202,11 +203,7 @@ def _get_parser():
         help='Path to the configuration file.',
     )
     traffic.add_argument(
-        '-o',
-        '--output-folder',
-        type=str,
-        required=False,
-        help='Output folder path.'
+        '-o', '--output-folder', type=str, required=False, help='Output folder path.'
     )
     traffic.add_argument(
         '-p',

github_analytics/drive.py

@@ -142,7 +142,7 @@ def get_or_create_gdrive_folder(parent_folder: str, folder_name: str) -> str:
 
     query = {
         'q': f"title = '{folder_name}' and mimeType = 'application/vnd.google-apps.folder' "
-             f"and '{parent_folder}' in parents and trashed = false"
+        f"and '{parent_folder}' in parents and trashed = false"
     }
     folders = drive.ListFile(query).GetList()
 
@@ -153,7 +153,7 @@ def get_or_create_gdrive_folder(parent_folder: str, folder_name: str) -> str:
     folder_metadata = {
         'title': folder_name,
         'mimeType': 'application/vnd.google-apps.folder',
-        'parents': [{'id': parent_folder}]
+        'parents': [{'id': parent_folder}],
     }
     folder = drive.CreateFile(folder_metadata)
     folder.Upload()

github_analytics/main.py

@@ -6,13 +6,13 @@
 
 import pandas as pd
 
+from github_analytics.drive import get_or_create_gdrive_folder
 from github_analytics.github.repository import RepositoryClient
 from github_analytics.github.repository_owner import RepositoryOwnerClient
 from github_analytics.github.traffic import TrafficClient
 from github_analytics.github.users import UsersClient
 from github_analytics.metrics import compute_metrics
 from github_analytics.output import create_spreadsheet, load_spreadsheet
-from github_analytics.drive import get_or_create_gdrive_folder
 
 LOGGER = logging.getLogger(__name__)
 
@@ -279,7 +279,7 @@ def collect_traffic(token, projects, output_folder):
         ouptut_folder (str):
             Folder in which the metrics will be stored.
     """
-    timestamp = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
+    timestamp = datetime.datetime.now().strftime('%Y-%m-%d_%H-%M-%S')
     for project, repositories in projects.items():
         for repository in repositories:
             repository_name = repository.split('/')[-1]
@@ -321,4 +321,4 @@ def collect_project_traffic(token, repository, repo_path):
         create_spreadsheet(f'gdrive://{repo_path}', traffic_data)
         return None
 
-    return sheets
+    return traffic_data

github_analytics/slack_utils.py

@@ -69,8 +69,8 @@ def post_slack_message_in_thread(channel, text, thread_ts):
     return response
 
 
-
 def send_alert(args):
+    """Send an alert message to a slack channel."""
     url = GITHUB_URL_PREFIX + args.run_id
     message = f'Github Analytics build failed :fire: :dumpster-fire: :fire: See errors <{url}|here>'
     post_slack_message(args.channel, message)