added slack notification

Author: FedeClaudi

proj/environment/manager.py

@@ -14,6 +14,7 @@
 from proj.animation.animate import animate_from_images
 from proj.plotting.results import plot_results
 from proj.utils.dropbox import DropBoxUtils, upload_folder
+from proj.utils.slack import send_slack_message
 
 FORMAT = "%(message)s"
 logging.basicConfig(
@@ -26,8 +27,9 @@ def __init__(self, model, winstor=False):
         """
             This class manages paths, saving of results etc.
         """
-        self.model = model
+        self.simstart = timestamp()
 
+        self.model = model
         self.exp_name = f'{model.trajectory["name"]}_{timestamp()}_{np.random.randint(low=0, high=10000)}'
 
         # get main folder
@@ -158,3 +160,16 @@ def conclude(self):
         # Upload results to dropbox
         if self.winstor:
             self._upload_to_dropbox()
+        else:
+            logging.info("Did not upload to dropbox")
+
+        logging.info("Sending slack message")
+        send_slack_message(
+            f"""
+            \n
+            Completed simulation
+            Start time: {self.simstart}
+            End time: {timestamp()}
+            Data folder: {self.datafolder}
+            """
+        )

proj/run/runner.py

@@ -92,8 +92,8 @@ def run_experiment(
             environment.itern = itern
             environment.update_world(g_xs, elapsed=itern * model.dt)
 
-            # log
-            if itern % 25 == 0:
+            # log status once a second
+            if itern % int(1 / model.dt) == 0:
                 log.info(
                     f"Iteration {itern}/{n_steps}. Current cost: {environment.curr_cost}."
                 )

proj/utils/slack.py

@@ -0,0 +1,13 @@
+from slack import WebClient
+from slack.errors import SlackApiError
+
+from proj.secrets import SLACK_TOKEN, SLACK_USER_ID
+
+
+def send_slack_message(message):
+    client = WebClient(token=SLACK_TOKEN)
+
+    try:
+        client.chat_postMessage(channel=SLACK_USER_ID, text=message)
+    except SlackApiError as e:
+        print(f"Got an error: {e.response['error']}")

setup.py

@@ -16,6 +16,7 @@
     "rich",
     "sklearn",
     "fancylog",
+    "dropbox",
 ]
 
 setup(

workspace.py

@@ -1,37 +0,0 @@
-# %%
-from proj.utils.dropbox import DropBoxUtils
-
-db = DropBoxUtils()
-
-
-# %%
-from pathlib import Path
-
-fld = Path("/Users/federicoclaudi/Documents/Github/pysical_locomotion/proj")
-
-
-# %%
-def upload_folder(dbx, fld, base):
-    fld = Path(fld)
-    base = Path(base)
-
-    # loop subfolders
-    for subf in fld.glob("*"):
-        if not subf.is_dir():
-            continue
-        upload_folder(dbx, subf, base / subf.name)
-
-    for f in fld.glob("*.*"):
-        if not f.is_file():
-            continue
-
-        dest = base / f.name
-        dbx.upload_file(f, dest)
-
-
-base = ""
-upload_folder(db, fld, "test")
-# %%
-db.upload_folder(fld, "test2")
-
-# %%