cleanup

slic/utils/elog.py

@@ -1,9 +1,26 @@
-import elog
-from getpass import getuser
-from getpass import getpass
 import os
-from .screenshot import Screenshot
 from pathlib import Path
+from getpass import getuser, getpass
+import elog
+from .screenshot import Screenshot
+
+
+class Elog:
+
+    def __init__(self, url, screenshot_directory="", **kwargs):
+        self.elog, self.user = get_default_elog_instance(url, **kwargs)
+        self.screenshot = Screenshot(screenshot_directory)
+        self.read = self.log.read
+
+    def post(self, *args, **kwargs):
+        if "Author" not in kwargs:
+            kwargs["Author"] = self.user
+        return self.log.post(*args, **kwargs)
+
+    def screenshot(self, message="", window=False, desktop=False, delay=3, **kwargs):
+        filepath = self.screenshot.shoot()[0]
+        kwargs["attachments"] = [filepath]
+        self.post(message, **kwargs)
 
 
 def get_default_elog_instance(url, **kwargs):
@@ -25,22 +42,4 @@ def get_default_elog_instance(url, **kwargs):
     return elog.open(url, **kwargs), kwargs["user"]
 
 
-class Elog:
-
-    def __init__(self, url, screenshot_directory="", **kwargs):
-        self.elog, self.user = get_default_elog_instance(url, **kwargs)
-        self.screenshot = Screenshot(screenshot_directory)
-        self.read = self.log.read
-
-    def post(self, *args, **kwargs):
-        if "Author" not in kwargs:
-            kwargs["Author"] = self.user
-        return self.log.post(*args, **kwargs)
-
-    def screenshot(self, message="", window=False, desktop=False, delay=3, **kwargs):
-        filepath = self.screenshot.shoot()[0]
-        kwargs["attachments"] = [filepath]
-        self.post(message, **kwargs)
-
-