make fmt (black 21.9b0)

1 files changed, 57 insertions, 55 deletions

diff --git a/python/pdftrio_tool.py b/python/pdftrio_tool.py
index dbe5b10..9d3010e 100755
--- a/python/pdftrio_tool.py
+++ b/python/pdftrio_tool.py
@@ -18,17 +18,15 @@ def run_classify_pdf_json(args):
     pdftrio_client = PdfTrioClient(host_url=args.pdftrio_host)
     wayback_client = WaybackClient()
     if args.jobs > 1:
-        worker = PdfTrioWorker(pdftrio_client,
-                               wayback_client,
-                               sink=None,
-                               mode=args.pdftrio_mode)
+        worker = PdfTrioWorker(
+            pdftrio_client, wayback_client, sink=None, mode=args.pdftrio_mode
+        )
         multi_worker = MultiprocessWrapper(worker, args.sink)
         pusher = JsonLinePusher(multi_worker, args.json_file, batch_size=args.jobs)
     else:
-        worker = PdfTrioWorker(pdftrio_client,
-                               wayback_client,
-                               sink=args.sink,
-                               mode=args.pdftrio_mode)
+        worker = PdfTrioWorker(
+            pdftrio_client, wayback_client, sink=args.sink, mode=args.pdftrio_mode
+        )
         pusher = JsonLinePusher(worker, args.json_file)
     pusher.run()
 
@@ -37,28 +35,26 @@ def run_classify_pdf_cdx(args):
     pdftrio_client = PdfTrioClient(host_url=args.pdftrio_host)
     wayback_client = WaybackClient()
     if args.jobs > 1:
-        worker = PdfTrioWorker(pdftrio_client,
-                               wayback_client,
-                               sink=None,
-                               mode=args.pdftrio_mode)
+        worker = PdfTrioWorker(
+            pdftrio_client, wayback_client, sink=None, mode=args.pdftrio_mode
+        )
         multi_worker = MultiprocessWrapper(worker, args.sink)
         pusher = CdxLinePusher(
             multi_worker,
             args.cdx_file,
             filter_http_statuses=[200, 226],
-            filter_mimetypes=['application/pdf'],
+            filter_mimetypes=["application/pdf"],
             batch_size=args.jobs,
         )
     else:
-        worker = PdfTrioWorker(pdftrio_client,
-                               wayback_client,
-                               sink=args.sink,
-                               mode=args.pdftrio_mode)
+        worker = PdfTrioWorker(
+            pdftrio_client, wayback_client, sink=args.sink, mode=args.pdftrio_mode
+        )
         pusher = CdxLinePusher(
             worker,
             args.cdx_file,
             filter_http_statuses=[200, 226],
-            filter_mimetypes=['application/pdf'],
+            filter_mimetypes=["application/pdf"],
         )
     pusher.run()
 
@@ -72,52 +68,58 @@ def run_classify_pdf_zipfile(args):
 
 def main():
     parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
-    parser.add_argument('--kafka-mode',
-                        action='store_true',
-                        help="send output to Kafka (not stdout)")
-    parser.add_argument('--kafka-hosts',
-                        default="localhost:9092",
-                        help="list of Kafka brokers (host/port) to use")
-    parser.add_argument('--kafka-env',
-                        default="dev",
-                        help="Kafka topic namespace to use (eg, prod, qa, dev)")
-    parser.add_argument('-j',
-                        '--jobs',
-                        default=8,
-                        type=int,
-                        help="parallelism for batch CPU jobs")
-    parser.add_argument('--pdftrio-host',
-                        default="http://pdftrio.qa.fatcat.wiki",
-                        help="pdftrio API host/port")
-    parser.add_argument('--pdftrio-mode',
-                        default="auto",
-                        help="which classification mode to use")
+    parser.add_argument(
+        "--kafka-mode", action="store_true", help="send output to Kafka (not stdout)"
+    )
+    parser.add_argument(
+        "--kafka-hosts",
+        default="localhost:9092",
+        help="list of Kafka brokers (host/port) to use",
+    )
+    parser.add_argument(
+        "--kafka-env", default="dev", help="Kafka topic namespace to use (eg, prod, qa, dev)"
+    )
+    parser.add_argument(
+        "-j", "--jobs", default=8, type=int, help="parallelism for batch CPU jobs"
+    )
+    parser.add_argument(
+        "--pdftrio-host", default="http://pdftrio.qa.fatcat.wiki", help="pdftrio API host/port"
+    )
+    parser.add_argument(
+        "--pdftrio-mode", default="auto", help="which classification mode to use"
+    )
     subparsers = parser.add_subparsers()
 
     sub_classify_pdf_json = subparsers.add_parser(
-        'classify-pdf-json',
-        help="for each JSON line with CDX info, fetches PDF and does pdftrio classify_pdfion")
+        "classify-pdf-json",
+        help="for each JSON line with CDX info, fetches PDF and does pdftrio classify_pdfion",
+    )
     sub_classify_pdf_json.set_defaults(func=run_classify_pdf_json)
-    sub_classify_pdf_json.add_argument('json_file',
-                                       help="JSON file to import from (or '-' for stdin)",
-                                       type=argparse.FileType('r'))
+    sub_classify_pdf_json.add_argument(
+        "json_file",
+        help="JSON file to import from (or '-' for stdin)",
+        type=argparse.FileType("r"),
+    )
 
     sub_classify_pdf_cdx = subparsers.add_parser(
-        'classify-pdf-cdx',
-        help="for each CDX line, fetches PDF and does pdftrio classify_pdfion")
+        "classify-pdf-cdx",
+        help="for each CDX line, fetches PDF and does pdftrio classify_pdfion",
+    )
     sub_classify_pdf_cdx.set_defaults(func=run_classify_pdf_cdx)
-    sub_classify_pdf_cdx.add_argument('cdx_file',
-                                      help="CDX file to import from (or '-' for stdin)",
-                                      type=argparse.FileType('r'))
+    sub_classify_pdf_cdx.add_argument(
+        "cdx_file",
+        help="CDX file to import from (or '-' for stdin)",
+        type=argparse.FileType("r"),
+    )
 
     sub_classify_pdf_zipfile = subparsers.add_parser(
-        'classify-pdf-zipfile',
-        help=
-        "opens zipfile, iterates over PDF files inside and does pdftrio classify_pdf for each")
+        "classify-pdf-zipfile",
+        help="opens zipfile, iterates over PDF files inside and does pdftrio classify_pdf for each",
+    )
     sub_classify_pdf_zipfile.set_defaults(func=run_classify_pdf_zipfile)
-    sub_classify_pdf_zipfile.add_argument('zip_file',
-                                          help="zipfile with PDFs to classify",
-                                          type=str)
+    sub_classify_pdf_zipfile.add_argument(
+        "zip_file", help="zipfile with PDFs to classify", type=str
+    )
 
     args = parser.parse_args()
     if not args.__dict__.get("func"):
@@ -133,5 +135,5 @@ def main():
     args.func(args)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()