Format chat folder (Mini-Conf#58)

* check chat folder as well

* format

* drive-by

* ok

* remove unused imports

* lint

* format

Author: hao-fang

Makefile

@@ -1,4 +1,4 @@
-PYTHON_FILES = main.py scripts/
+PYTHON_FILES = main.py scripts/ chat/
 JS_FILES = $(shell find static/js -name "*.js")
 CSS_FILES = $(shell find static/css -name "*.css")
 .PHONY: format-python format-web format run freeze format-check

chat/make_poster_rooms.py

@@ -1,23 +1,22 @@
 import argparse
 import csv
-import random
-import string
+import json
 
+import yaml
 from requests import sessions
-from pprint import pprint
 from rocketchat_API.rocketchat import RocketChat
-import yaml
-import json
+
 
 def parse_arguments():
     parser = argparse.ArgumentParser(description="MiniConf Portal Command Line")
     parser.add_argument("--config", default="config.yml", help="Configuration yaml")
     parser.add_argument("--papers", default="../sitedata/papers.csv", help="Papers CSV")
-    parser.add_argument("--test", action='store_true')
+    parser.add_argument("--test", action="store_true")
     return parser.parse_args()
 
+
 def read_papers(fname):
-    name, typ = fname.split("/")[-1].split(".")
+    _name, typ = fname.split("/")[-1].split(".")
     if typ == "json":
         res = json.load(open(fname))
     elif typ in {"csv", "tsv"}:
@@ -28,36 +27,33 @@ def read_papers(fname):
         raise ValueError("file not supported: " + fname)
     return res
 
+
 if __name__ == "__main__":
     args = parse_arguments()
 
-
     config = yaml.load(open(args.config))
     papers = read_papers(args.papers)
 
     with sessions.Session() as session:
-        rocket = RocketChat(config["username"],
-                            config["password"],
-                            server_url=config["server"],
-                            session=session)
+        rocket = RocketChat(
+            config["username"],
+            config["password"],
+            server_url=config["server"],
+            session=session,
+        )
 
         for paper in papers:
             channel_name = "paper_" + paper["UID"]
             if not args.test:
                 created = rocket.channels_create(channel_name).json()
                 print(channel_name, created)
-            id = rocket.channels_info(channel = channel_name).json()["channel"]["_id"]
-
-            
-            # Change to topic of papers. 
-            topic = "%s - %s"%(
-                paper["title"],
-                paper["authors"],
-            )
+            channel_id = rocket.channels_info(channel=channel_name).json()["channel"][
+                "_id"
+            ]
+
+            # Change to topic of papers.
+            topic = "%s - %s" % (paper["title"], paper["authors"],)
             if not args.test:
-                rocket.channels_set_topic(id, topic).json()
+                rocket.channels_set_topic(channel_id, topic).json()
 
             print("Creating " + channel_name + " topic " + topic)
-
-
-

main.py

@@ -84,7 +84,7 @@ def papers():
 
 
 @app.route("/paper_vis.html")
-def paperVis():
+def paper_vis():
     data = _data()
     return render_template("papers_vis.html", **data)
 

mypy.ini

@@ -6,3 +6,6 @@ strict_optional = False
 [mypy-sklearn.*,transformers.*,flask_frozen.*,flaskext.*,ics.*]
 ignore_missing_imports = True
 
+[mypy-rocketchat_API.*]
+ignore_missing_imports = True
+

requirements-dev.txt

@@ -2,3 +2,4 @@ black==19.10b0
 pylint==2.4.4
 mypy==0.761
 -r scripts/requirements.txt
+-r chat/requirements.txt