Merge branch 'bnewbold-review-bots'

1 files changed, 60 insertions, 0 deletions

diff --git a/python/fatcat_review.py b/python/fatcat_review.py
new file mode 100755
index 00000000..40bc7041
--- /dev/null
+++ b/python/fatcat_review.py
@@ -0,0 +1,60 @@
+#!/usr/bin/env python3
+
+import sys
+import argparse
+import datetime
+import raven
+
+from fatcat_tools import authenticated_api
+from fatcat_tools.reviewers import DummyReviewBot, ReviewBot
+
+# Yep, a global. Gets DSN from `SENTRY_DSN` environment variable
+sentry_client = raven.Client()
+
+
+def run_dummy(args):
+    reviewer = DummyReviewBot(args.api, poll_interval=args.poll_interval,
+        verbose=args.debug)
+    if args.editgroup:
+        annotation = reviewer.run_single(args.editgroup, args.annotate)
+        print(annotation)
+    elif args.continuous:
+        reviewer.run()
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--debug',
+        action='store_true',
+        help="enable debug logging")
+    parser.add_argument('--api-host-url',
+        default="http://localhost:9411/v0",
+        help="fatcat API host/port to use")
+    parser.add_argument('--poll-interval',
+        help="how long to wait between polling (seconds)",
+        default=10.0, type=float)
+    subparsers = parser.add_subparsers()
+
+    sub_dummy = subparsers.add_parser('dummy')
+    sub_dummy.set_defaults(func=run_dummy)
+    sub_dummy.add_argument("--continuous",
+        action="store_true",
+        help="run forever, polling for new reviewable editgroups")
+    sub_dummy.add_argument("--editgroup",
+        help="single editgroup ID to review")
+    sub_dummy.add_argument("--annotate",
+        action="store_true",
+        help="for single editgroups, pushes result as annotation")
+
+    args = parser.parse_args()
+    if not args.__dict__.get("func"):
+        print("tell me what to do!")
+        sys.exit(-1)
+    if (args.editgroup and args.continuous) or not (args.editgroup or args.continuous):
+        print("need to run on a single editgroup, or continuous")
+        sys.exit(-1)
+
+    args.api = authenticated_api(args.api_host_url)
+    args.func(args)
+
+if __name__ == '__main__':
+    main()