def startScript(self, script):
script.register(self)
for (packet, handler) in script.handlers():
self.pm.registerHandler(self.mapSocket, packet, handler)
self.scriptInstances.append(script)
script.start()
def test_start(self, mock_socket, mock_send_tweets_to_spark,
mock_get_tweets):
mock_socket.bind.return_value = mock.Mock()
mock_socket.accept.return_value = mock.Mock()
start()
assert mock_get_tweets.called_once()
assert mock_send_tweets_to_spark.called_once()
def startScript(self, script):
script.register(self)
for (packet, handler) in script.handlers():
self.pm.registerHandler(self.mapSocket, packet, handler)
self.scriptInstances.append(script)
script.start()
import sys
sys.path = ['', 'pyglet-1.1.4'] + sys.path
import pyglet
window = pyglet.window.Window(width=800,
height=600,
caption="You are the Hero")
from pyglet.window import key
from scene import Scene
import script
scene = Scene(window.width, window.height)
script.start(scene)
keymap = {
key.E: "up",
key.S: "left",
key.D: "down",
key.F: "right",
key.UP: "up",
key.DOWN: "down",
key.LEFT: "left",
key.RIGHT: "right",
key.SPACE: "blink",
}
pyglet.gl.glClearColor(255, 255, 255, 255)
def script(dirname, subdirname):
unzip(dirname, subdirname)
if not os.path.isdir('check/{}'.format(dirname)):
os.mkdir('check/{}'.format(dirname))
if not os.path.isdir('/mnt/volume-nyc1-01/{}'.format(dirname)):
os.mkdir('/mnt/volume-nyc1-01/{}'.format(dirname))
if not os.path.isdir(os.path.join('check', dirname, 'xlses')):
os.mkdir(os.path.join('check', dirname, 'xlses'))
if not os.path.isdir(os.path.join('check', dirname, 'xlses', subdirname)):
os.mkdir(os.path.join('check', dirname, 'xlses', subdirname))
try:
if os.path.isfile(
os.path.join('/root/Novosad/mouses/Data', dirname,
subdirname + '.zip')):
ext = '.zip'
if os.path.isfile(
os.path.join('/root/Novosad/mouses/Data', dirname,
subdirname + '.rar')):
ext = '.rar'
shutil.move(
os.path.join('/root/Novosad/mouses/Data', dirname,
subdirname + ext),
os.path.join('/mnt/volume-nyc1-01/', dirname))
except Exception as e:
print(e, 'already moved', subdirname)
part(dirname, subdirname)
start(dirname, subdirname)
time.sleep(20)
try:
while True:
if len([
f for f in os.listdir(
os.path.join('check', dirname, subdirname))
if 'done.txt' in f
]) == N_THREADS:
xls_files = []
for numb in range(N_THREADS):
xls_files += [
os.path.join('check', dirname, subdirname, str(numb),
f) for f in os.listdir(
os.path.join('check', dirname,
subdirname, str(numb)))
if '.xls' in f
]
for xls in xls_files:
shutil.move(
xls, os.path.join('check', dirname, 'xlses',
subdirname))
break
else:
time.sleep(60 * 1)
except Exception as e:
print(e)
clear(dirname, subdirname)
add_empt(dirname, subdirname)
create(dirname, subdirname)
concatenate('0', dirname, subdirname)
concatenate('1', dirname, subdirname)
infected = []
if os.path.isdir(os.path.join('check', dirname, 'xlses', 'avg', '144')):
infected = [
f for f in os.listdir(
os.path.join('check', dirname, 'xlses', 'avg', '144'))
if '.xls' in f
]
infected = list(set([f[0] for f in infected]))
for m in infected:
m = '0' if m == 'l' else '1'
test(m, dirname)
import sys
sys.path = ['', 'pyglet-1.1.4'] + sys.path
import pyglet
window = pyglet.window.Window(width=800, height=600,
caption = "You are the Hero")
from pyglet.window import key
from scene import Scene
import script
scene = Scene(window.width, window.height)
script.start(scene)
keymap = {
key.E : "up",
key.S : "left",
key.D : "down",
key.F : "right",
key.UP: "up",
key.DOWN: "down",
key.LEFT: "left",
key.RIGHT: "right",
key.SPACE : "blink",
}
pyglet.gl.glClearColor(255, 255, 255, 255)
