whitespace

Author: krivard

src/acquisition/covidcast/delete_batch.py

@@ -11,16 +11,16 @@
 
 
 def get_argument_parser():
-  """Define command line arguments."""
+    """Define command line arguments."""
 
-  parser = argparse.ArgumentParser()
-  parser.add_argument(
-    '--deletion_dir',
-    help='directory where deletion CSVs are stored')
-  parser.add_argument(
-    '--log_file',
-    help="filename for log output (defaults to stdout)")
-  return parser
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+      '--deletion_dir',
+      help='directory where deletion CSVs are stored')
+    parser.add_argument(
+      '--log_file',
+      help="filename for log output (defaults to stdout)")
+    return parser
 
 def handle_file(deletion_file, database):
     logger.info("Deleting from csv file", filename=deletion_file)
@@ -39,23 +39,23 @@ def handle_file(deletion_file, database):
     return 0
 
 def main(args):
-  """Delete rows from covidcast."""
+    """Delete rows from covidcast."""
 
-  logger = get_structured_logger("csv_deletion", filename=args.log_file)
-  start_time = time.time()
-  database = Database()
-  database.connect()
-  all_n = 0
+    logger = get_structured_logger("csv_deletion", filename=args.log_file)
+    start_time = time.time()
+    database = Database()
+    database.connect()
+    all_n = 0
 
-  try:
-      for deletion_file in sorted(glob.glob(os.path.join(args.deletion_dir, '*.csv'))):
-          all_n += handle_file(deletion_file)
-  finally:
-      database.disconnect(True)
+    try:
+        for deletion_file in sorted(glob.glob(os.path.join(args.deletion_dir, '*.csv'))):
+            all_n += handle_file(deletion_file)
+    finally:
+        database.disconnect(True)
 
-  logger.info(
-      "Deleted CSVs from database",
-      total_runtime_in_seconds=round(time.time() - start_time, 2), row_count=all_n)
+    logger.info(
+        "Deleted CSVs from database",
+        total_runtime_in_seconds=round(time.time() - start_time, 2), row_count=all_n)
 
 if __name__ == '__main__':
-  main(get_argument_parser().parse_args())
+    main(get_argument_parser().parse_args())