Split status and players to separate classes

1 files changed, 58 insertions, 0 deletions

diff --git a/mcwrapper/status.py b/mcwrapper/status.py
new file mode 100644
index 0000000..42f8ed8
--- /dev/null
+++ b/mcwrapper/status.py
@@ -0,0 +1,58 @@
+# vim: expandtab ft=python ts=4 sw=4 sts=4:
+import os
+
+from . import prints
+
+__STATUSSTRINGS__ = {
+    0: "Not running",
+    1: "Starting",
+    2: "Running",
+    3: "Stopping",
+    }
+__STATUSFILE__ = 'status'
+
+
+class MCStatus:
+    "Tracks server status"
+    def __init__(self, wrapper, file_export=False):
+        self.wrapper = wrapper
+        self.status = 0
+        wrapper.hook_start(self.__server_start__)
+        wrapper.hook_stop(self.__server_stop__)
+        wrapper.hook_line(': Done', self.__server_started__)
+        wrapper.hook_line(': Stopping the server', self.__server_stopping__)
+        self.file_export = file_export
+        if file_export:
+            with open(__STATUSFILE__, 'w') as file:
+                file.write(__STATUSSTRINGS__[0] + '\n')
+
+    def clean(self):
+        try:
+            os.remove(__STATUSFILE__)
+        except FileNotFoundError:
+            pass
+
+    def __server_start__(self):
+        self.status = 1
+        if self.file_export:
+            with open(__STATUSFILE__, 'w') as file:
+                file.write(__STATUSSTRINGS__[1] + '\n')
+
+    def __server_stop__(self):
+        if self.file_export:
+            with open(__STATUSFILE__, 'w') as file:
+                file.write(__STATUSSTRINGS__[0] + '\n')
+
+    def __server_started__(self, line):
+        prints.info("Server start.")
+        self.status = 2
+        if self.file_export:
+            with open(__STATUSFILE__, 'w') as file:
+                file.write(__STATUSSTRINGS__[2] + '\n')
+
+    def __server_stopping__(self, line):
+        prints.info("Server stop.")
+        self.status = 3
+        if self.file_export:
+            with open(__STATUSFILE__, 'w') as file:
+                file.write(__STATUSSTRINGS__[3] + '\n')