From bdb289d83bd1c86016ef2621b26cbc9c4a50bb6a Mon Sep 17 00:00:00 2001
From: Calvin McLean <calvinmclean@email.arizona.edu>
Date: Tue, 17 Sep 2019 08:34:39 -0700
Subject: [PATCH] Fix formatting

---
 .../enforce_special_allocation_quota.py       | 38 +++++++++++++------
 1 file changed, 26 insertions(+), 12 deletions(-)

diff --git a/scripts/jetstream/enforce_special_allocation_quota.py b/scripts/jetstream/enforce_special_allocation_quota.py
index a35b3210e..16ff0106d 100644
--- a/scripts/jetstream/enforce_special_allocation_quota.py
+++ b/scripts/jetstream/enforce_special_allocation_quota.py
@@ -15,19 +15,31 @@ def main():
     Reset quotas for users on a particular allocation source
     """
     parser = argparse.ArgumentParser()
-    parser.add_argument('--dry-run', action='store_true', help='Print output rather than perform operation')
-    parser.add_argument('--allocation-source', required=True,
-                        help='Allocation source name to reset quotas for, e.g. TG-ASC160018')
-    parser.add_argument('--quota-id', required=True, type=int,
-                        help='Quota ID to set')
-    parser.add_argument('--whitelist-quota-ids', type=lambda s: [int(item.strip()) for item in s.split(',')],
-                        help='Quota IDs that are acceptable and won\'t be overwritten (comma separated)')
+    parser.add_argument(
+        '--dry-run',
+        action='store_true',
+        help='Print output rather than perform operation'
+    )
+    parser.add_argument(
+        '--allocation-source',
+        required=True,
+        help='Allocation source name to reset quotas for, e.g. TG-ASC160018'
+    )
+    parser.add_argument(
+        '--quota-id', required=True, type=int, help='Quota ID to set'
+    )
+    parser.add_argument(
+        '--whitelist-quota-ids',
+        type=lambda s: [int(item.strip()) for item in s.split(',')],
+        help=
+        'Quota IDs that are acceptable and won\'t be overwritten (comma separated)'
+    )
     parser.set_defaults(dry_run=False)
     args = parser.parse_args()
-    return run_command(args.dry_run,
-                       args.allocation_source,
-                       args.quota_id,
-                       args.whitelist_quota_ids)
+    return run_command(
+        args.dry_run, args.allocation_source, args.quota_id,
+        args.whitelist_quota_ids
+    )
 
 
 def run_command(dry_run, allocation_source, quota_id, whitelist_quota_ids):
@@ -70,7 +82,9 @@ def run_command(dry_run, allocation_source, quota_id, whitelist_quota_ids):
             print('DRY-RUN: Not changing quota to {}'.format(quota_id))
         else:
             print('Changing provider quota to {}...'.format(quota_id))
-            updated_provider_quota = set_provider_quota(identity.uuid, quota=quota_to_set)
+            updated_provider_quota = set_provider_quota(
+                identity.uuid, quota=quota_to_set
+            )
             print('Updated provider quota: {}'.format(updated_provider_quota))
             print('Changing DB quota to {}...'.format(quota_id))
             identity.quota = quota_to_set