lint

Author: amontanez24

scripts/create_download_tracker.py

@@ -4,34 +4,35 @@
 
 import boto3
 import build
-import rdt
 import tomli
 import tomli_w
 
+import rdt
 
 PROJECT_PATH = './scripts/rdt-download-tracker'
 BUCKET = os.getenv('DOWNLOAD_TRACKER_BUCKET', '')
 S3_PACKAGE_PATH = 'simple/rdt-download-tracker/'
 
 
-def set_version(version):
+def _set_version(version):
     toml_path = os.path.join(PROJECT_PATH, 'pyproject.toml')
     with open(toml_path, 'rb') as f:
         pyproject = tomli.load(f)
         pyproject['project']['version'] = version
-    
+
     with open(toml_path, 'wb') as f:
         tomli_w.dump(pyproject, f)
-        
 
 
 def build_package():
-    set_version(rdt.__version__)
+    """Builds the wheel and sdist for 'rdt-download-tracker'."""
+    _set_version(rdt.__version__)
     build.ProjectBuilder(PROJECT_PATH).build('wheel', 'dist')
     build.ProjectBuilder(PROJECT_PATH).build('sdist', 'dist')
 
 
 def upload_package():
+    """Uploads the built package to the S3 bucket."""
     s3_client = boto3.client('s3')
     for file_name in os.listdir('dist'):
         dest = f'{S3_PACKAGE_PATH}/{file_name}'
@@ -40,4 +41,4 @@ def upload_package():
 
 if __name__ == '__main__':
     build_package()
-    upload_package()
+    upload_package()