blob: 8dc8b85fe6faf3be4def71bfce26657162f4a336 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
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.clear()
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
|
