summaryrefslogtreecommitdiff
path: root/jfr_playoff
diff options
context:
space:
mode:
authoremkael <emkael@tlen.pl>2017-08-01 00:38:10 +0200
committeremkael <emkael@tlen.pl>2017-08-01 00:38:10 +0200
commitb1bf9357d573fe4a03d98ea1da7e01c292233c5d (patch)
tree24beffa6f9ef9299c9517ca2e769ce1651ea0f25 /jfr_playoff
parentbd5fd6a6596dd7625019c6c0f0ff043597cda837 (diff)
File and Goniec operations moved to separate class
Diffstat (limited to 'jfr_playoff')
-rw-r--r--jfr_playoff/filemanager.py47
1 files changed, 47 insertions, 0 deletions
diff --git a/jfr_playoff/filemanager.py b/jfr_playoff/filemanager.py
new file mode 100644
index 0000000..64ac6d9
--- /dev/null
+++ b/jfr_playoff/filemanager.py
@@ -0,0 +1,47 @@
+import os, shutil, socket
+import __main__
+
+class PlayoffFileManager(object):
+
+ def __init__(self, settings):
+ self.settings = settings
+ self.goniec = self.settings.get('goniec')
+ self.output_file = self.settings.get('output')
+ self.output_path = os.path.dirname(self.output_file).strip(os.sep) + os.sep
+ self.files = set()
+
+ def reset(self):
+ self.files.empty()
+
+ def register_file(self, path):
+ if path.startswith(self.output_path):
+ self.files.add(path.replace(self.output_path, ''))
+
+
+ def write_content(self, content):
+ output = open(self.output_file, 'w')
+ output.write(content.encode('utf8'))
+ output.close()
+ self.register_file(self.output_file)
+ return self.output_file
+
+ def copy_scripts(self, script_path='sklady/playoff.js'):
+ script_output_path = os.path.join(self.output_path, script_path)
+ shutil.copy(
+ unicode(os.path.join(os.path.dirname(__main__.__file__), 'playoff.js')),
+ unicode(script_output_path)
+ )
+ self.register_file(script_output_path)
+ return script_output_path
+
+ def send_files(self):
+ if self.goniec['enabled']:
+ try:
+ content_lines = [self.output_path] + list(self.files) + ['bye', '']
+ print '\n'.join(content_lines)
+ goniec = socket.socket()
+ goniec.connect((self.goniec['host'], self.goniec['port']))
+ goniec.sendall('\n'.join(content_lines))
+ goniec.close()
+ except socket.error:
+ pass