1 files changed, 86 insertions, 0 deletions
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..9518267
--- /dev/null
+++ b/main.py
@@ -0,0 +1,86 @@
+import sys  # read config from args
+import json
+import time
+import logging
+import unittest
+import test
+
+
+from slack_parser import SlackParser
+from msgraph import MSGraphApi
+
+# Optional logging
+logging.basicConfig(level=logging.INFO)
+LIMIT_THROUGHPUT = False
+
+if __name__ == '__main__':
+
+    # run the tests first
+    suite = unittest.TestLoader().loadTestsFromModule(test)
+    result = unittest.TextTestRunner(verbosity=0).run(suite)
+    if result.failures != []:
+        logging.error("1 or more tests failed. Exiting with code 1.")
+        sys.exit(1)
+
+    config = json.load(open(sys.argv[1]))
+
+    #msapi = MSGraphApi(config).getAuth() # "app flow"
+    msapi = MSGraphApi(config, flow="obo").getAuth()
+
+    if msapi is not None:
+        # connection acquired, begin doing stuff
+
+        # Example Calls
+        teamId = msapi.getTeamByName(config["teamname"])
+        channelId = msapi.getChatByNameAndTeamId(teamId, config["channelname"])
+
+        #graph_data = msapi.getUsers().json()
+        #graph_data = msapi.getChatByTeamAndId(teamId, channelId).json()
+        #graph_data = msapi.writeChatMessageByTeamAndId(
+        #        teamId=teamId,
+        #        channelId=channelId,
+        #        payload={ "body" : { "content" : "hello world" } }
+        #).json()
+
+        #print("Graph API call result: ")
+        #print(json.dumps(graph_data, indent=2))
+        #sys.exit(0)
+
+        sp = SlackParser("resources/channels/devops", "resources/users.json", ["text", "ts", "name"]).buildUsersLibrary()
+        if sp.usersLib is {}:
+            logging.error("Failed to read users. Shutting down with code 1.")
+            sys.exit(1)
+
+        stats_mssg_counter = 1
+        logging.info("LIMIT_THROUGHPUT: " + str(LIMIT_THROUGHPUT))
+
+        # load 1 json file at a time in memory
+        for slackDay in sp.getSortedMessageFiles():
+            # parse and write 1 json message at a time
+            for message in sp.getMessages(slackDay):
+                reduced_message = sp.getKeyValues(message)
+                if reduced_message is not None:
+                    teams_mssg = msapi.buildTeamsMessage(reduced_message)
+                    graph_data = msapi.writeChatMessageByTeamAndId(
+                            teamId=teamId,
+                            channelId=channelId,
+                            payload=teams_mssg
+                    )
+                    if graph_data.status_code != 201:
+                        logging.error("failed to write to Teams")
+                        logging.error(teams_mssg)
+                        logging.error(graph_data.json())
+                    else:
+                        logging.info("success")
+                stats_mssg_counter += 1
+
+                # control the throughput
+                if LIMIT_THROUGHPUT:
+                    if stats_mssg_counter % 10 == 0:
+                        time.sleep(10)
+                    else:
+                        time.sleep(1)
+
+    else:
+        logging.error("Failed to Authenticate with Ms Graph. Shutting down with code 1.")
+        sys.exit(1)