def getScripts():
global lastUpdate
print("Grabbing more scripts...")
info = reddit.getInfo('AskReddit', settings.reddit_amount_posts)
new_scripts = len([script for script in info if not script.update])
updating_scripts = len([script for script in info if script.update])
print("Adding %s new scripts, updating %s" %
(new_scripts, updating_scripts))
for script in info:
if script.update:
database.updateSubmission(script)
else:
database.addSubmission(script)
lastUpdate = datetime.datetime.now()
def clientTick(client):
print("Server tick thread started for client")
HEADERSIZE = 10
while True:
if client.disconnect:
print("%s SERVER user %s disconnected" %
(datetime.datetime.now(), repr(client.username)))
break
full_msg = b''
new_msg = True
while True:
try:
client_connection = client.connection
buf = client_connection.recv(2048)
if new_msg:
try:
msglen = int(buf[:HEADERSIZE])
except ValueError:
print("client disconnect error")
# happens when client disconnects
break
new_msg = False
full_msg += buf
except ConnectionResetError:
print("%s SERVER user %s connecton reset error" %
(datetime.datetime.now(), repr(client.username)))
break
download_size = len(full_msg) - HEADERSIZE
if download_size == msglen:
if download_size > 100000:
print("%s SERVER received large message (%s)" %
(datetime.datetime.now(),
str(download_size / 1000000) + "MB"))
try:
incomingdata = pickle.loads(full_msg[HEADERSIZE:])
except EOFError:
print("%s SERVER user %s disconnected" %
(datetime.datetime.now(), repr(client.username)))
break
new_msg = True
full_msg = b""
if not client.authorized:
if "login-attempt" == incomingdata[0]:
print("%s SERVER user %s login attempt" %
(datetime.datetime.now(), repr(incomingdata[1])))
username = incomingdata[1]
password = incomingdata[2]
login = (database.login(username, password))
online_users = database.getOnlineUsers()
if username in online_users:
print("%s SERVER user %s already logged in" %
(datetime.datetime.now(),
repr(incomingdata[1])))
sendToClient(client_connection,
("login-success", False, None))
else:
if login:
key = generateKey()
client.key = key
client.username = username
sendToClient(client_connection,
("login-success", True, key))
client.authorized = True
print("%s SERVER user %s logged in" %
(datetime.datetime.now(),
repr(incomingdata[1])))
database.updateUserStatus(username, "ONLINE")
else:
sendToClient(client_connection,
("login-success", False, None))
print("%s SERVER user %s wrong password" %
(datetime.datetime.now(),
repr(incomingdata[1])))
else:
if "request-scripts" == incomingdata[1]:
print("%s SERVER user %s request scripts" %
(datetime.datetime.now(), repr(client.username)))
if incomingdata[0] == client.key:
print("%s SERVER sending scripts to user %s" %
(datetime.datetime.now(),
repr(client.username)))
amount = incomingdata[2]
filter = incomingdata[3]
if filter == "ups":
data = database.getScripts(amount, "ups")
sendToClient(client_connection,
("scripts-return", data,
settings.music_types))
elif filter == "latest posts":
data = database.getScripts(
amount, "timecreated")
sendToClient(client_connection,
("scripts-return", data,
settings.music_types))
elif filter == "recently added":
data = database.getScripts(
amount, "timegathered")
sendToClient(client_connection,
("scripts-return", data,
settings.music_types))
elif filter == "comments":
data = database.getScripts(
amount, "num_comments")
sendToClient(client_connection,
("scripts-return", data,
settings.music_types))
pass
else:
print("%s SERVER user %s key does not match up" %
(datetime.datetime.now(),
repr(client.username)))
elif "edit-script" == incomingdata[1]:
scriptno = incomingdata[2]
print("%s SERVER user %s request to edit script %s" %
(datetime.datetime.now(), repr(
client.username), scriptno))
if incomingdata[0] == client.key:
script_status = database.getScriptStatus(scriptno)
if script_status == "RAW":
print(
"%s SERVER allowing user %s to edit script %s"
% (datetime.datetime.now(),
repr(client.username), scriptno))
client.editingScript = scriptno
database.updateScriptStatus(
"EDITING", client.username, scriptno)
sendToClient(
client.connection,
('edit-script-success', True, scriptno))
sendToAllClients(
('script-status-update', scriptno,
"EDITING", client.username))
print(
"%s SERVER sending all clients (%s) status update for %s"
%
(datetime.datetime.now(),
len(getAllClientConnections()), scriptno))
elif script_status == "EDITING":
print(
"%s SERVER refusing user %s to edit script %s"
% (datetime.datetime.now(),
repr(client.username), scriptno))
sendToClient(
client.connection,
('edit-script-success', False, scriptno))
else:
print("%s SERVER user %s key does not match up" %
(datetime.datetime.now(),
repr(client.username)))
elif "upload-video" == incomingdata[1]:
if incomingdata[0] == client.key:
scriptno = incomingdata[2]
video_generator_payload = incomingdata[3]
script_status = database.getScriptStatus(scriptno)
if script_status == "EDITING":
if scriptno == client.editingScript:
print(
"%s SERVER allowing user %s to upload script number %s"
% (datetime.datetime.now(),
repr(client.username), scriptno))
if database.uploadVid(
video_generator_payload, scriptno):
database.updateScriptStatus(
"COMPLETE", client.username,
scriptno)
sendToClient(client_connection,
('script-upload-success',
True, scriptno))
client.scriptsComplete.append(scriptno)
client.editingScript = None
else:
sendToClient(client_connection,
('script-upload-success',
False, scriptno))
sendToAllClients(
('script-status-update', scriptno,
"COMPLETE", client.username))
else:
print(
"%s SERVER user %s script number %s does not match what client is editing %s"
% (datetime.datetime.now(),
repr(client.username), scriptno,
client.editingScript))
else:
print("%s SERVER user %s script status is %s" %
(datetime.datetime.now(),
repr(client.username), script_status))
else:
print("%s SERVER user %s key does not match up" %
(datetime.datetime.now(),
repr(client.username)))
elif "quit-editing" == incomingdata[1]:
if incomingdata[0] == client.key:
scriptno = incomingdata[2]
if client.editingScript == scriptno:
database.updateScriptStatus(
"RAW", None, scriptno)
print("%s SERVER user %s quit editing %s" %
(datetime.datetime.now(),
repr(client.username), scriptno))
sendToAllClients(('script-status-update',
scriptno, "RAW", None))
client.editingScript = None
else:
print(
"%s SERVER user %s not editing script %s" %
(datetime.datetime.now(),
repr(client.username), scriptno))
else:
print("%s SERVER user %s key does not match up" %
(datetime.datetime.now(),
repr(client.username)))
elif "flag-scripts" == incomingdata[1]:
if incomingdata[0] == client.key:
scriptno = incomingdata[2]
flagtype = incomingdata[3]
database.updateScriptStatus(
flagtype, client.username, scriptno)
print(
"%s SERVER user %s flagging script %s as %s" %
(datetime.datetime.now(), repr(
client.username), scriptno, flagtype))
sendToAllClients(('script-status-update', scriptno,
flagtype, client.username))
client.editingScript = None
else:
print("%s SERVER user %s key does not match up" %
(datetime.datetime.now(),
repr(client.username)))
elif "add-script" == incomingdata[1]:
if incomingdata[0] == client.key:
url = incomingdata[2]
try:
post = reddit.getPostByUrl(url)
if post is not None:
print("%s SERVER user %s added script %s" %
(datetime.datetime.now(),
repr(client.username),
post.submission_id))
database.addSubmission(post)
sendToClient(client_connection,
('add-script-success', True,
"Successfully added script"))
else:
print(
"%s SERVER user %s attempted to add script that already exists"
% (datetime.datetime.now(),
repr(client.username)))
sendToClient(
client_connection,
('add-script-success', False,
"Script already in database"))
except Exception as e:
print(
"%s SERVER user %s error attempting to add script %s"
% (datetime.datetime.now(),
repr(client.username), url))
sendToClient(client_connection, (
'add-script-success', False,
"An error occured trying to add the script"
))
else:
print("%s SERVER user %s key does not match up" %
(datetime.datetime.now(),
repr(client.username)))
elif "PING" == incomingdata[1]:
if incomingdata[0] == client.key:
client.lastPing = datetime.datetime.now()
print("%s SERVER sending PONG to %s" %
(datetime.datetime.now(),
repr(client.username)))
sendToClient(client.connection, ('PONG', ))
else:
print("%s SERVER user %s key does not match up" %
(datetime.datetime.now(),
repr(client.username)))
if (datetime.datetime.now().minute -
client.lastPing.minute) > 2:
print(
"%s SERVER no PING from %s in 2 minutes. Disconnecting"
% (datetime.datetime.now(), repr(client.username)))
client.disconnect = True
print("%s SERVER Thread shutting down" % datetime.datetime.now())
client.disconnect = True
break