def UpdateUserPoints(UserID): # Start user point count UserPoints = 0 # Query all complete unique solutions challenges = Challenges.QueryAllChallenges() for challenge in challenges: if Solutions.HasUserSolvedChallenge(UserID, challenge.ChallengeID): UserPoints += Challenges.ChallengeQueryID(challenge.ChallengeID).Points # Query all achievements achievements = Achievements.AchievementsQueryUserID(UserID) for achievement in achievements: UserPoints += int(achievement.AchievementScore) # Update this user's score in the database ExistingUsers = Session.query(UsersTable).filter(UsersTable.UserID == UserID).all() # Commit changes if len(ExistingUsers) == 1: ExistingUsers[0].UserPoints = UserPoints Session.commit() # Post to session if we have a matching ID (otherwise, just return the user points) if session.get("UserID") and session["UserID"] == UserID: session["UserPoints"] = UserPoints session.save() return UserPoints
def UserDelete(UserID, Password, ConfirmPassword): # Is the form empty? Return no error... if not UserID and not Password and not ConfirmPassword: return "" # Do these user passwords match? if Password != ConfirmPassword: return "Error: Passwords do not match!" # Get existing user ExistingUsers = Session.query(UsersTable).filter(UsersTable.UserID == UserID).all() if len(ExistingUsers) <= 0: return "Error: User name does not exist." ExistingUser = ExistingUsers[0] # Is the given password the active password? ExistingHash = __GeneratePasswordHash(Password) if ExistingUser.UserPassword != ExistingHash: return "Error: Given password is not correct." # Delete user from users table Session.delete(ExistingUser) Session.commit() # Logout user UserLogout() # Done! return "deleted"
def UserSetIconID(UserID, IconID): # Convert to integer if not IconID.isdigit(): return "Error: IconID is not an integer." else: IconID = int(IconID) # Bounds check the icon ID (should only range betwee [0, 18] if IconID < 0 or IconID > 19: return "Error: IconID is out of range." # Update this user's score in the database ExistingUsers = Session.query(UsersTable).filter(UsersTable.UserID == UserID).all() # Commit changes if len(ExistingUsers) == 1: ExistingUsers[0].IconID = IconID Session.commit() else: return "Error: No users found." # Post to session if we have a matching ID (otherwise, just return the user points) if session.get("UserID") and session["UserID"] == UserID: session["UserIconID"] = IconID session.save() # All done! return "Icon change saved!"
def UserReset(UserName): # Get existing user ExistingUsers = Session.query(UsersTable).filter(UsersTable.UserName == UserName).all() if len(ExistingUsers) <= 0: return "Error: User name does not exist." # Register this new user into the database ExistingUser = ExistingUsers[0] # Randomize an 8-character string NewPassword = "" for i in range(8): NewPassword += random.choice(h.string_whitelist) # Send an email with the password try: # Form a message message = """\ From: [email protected] To: %s Subject: Password Change New Password: \"%s\" (Change this as soon as possible!) Visit www.corecodex.com to learn more. """ % ( ExistingUser.UserEMail, NewPassword, ) # Send mail s = smtplib.SMTP() s.connect() s.sendmail("*****@*****.**", ExistingUser.UserEMail, message) s.quit() except: return "Error: Unable to send email. Password was not reset." # Hash password ExistingUser.UserPassword = __GeneratePasswordHash(NewPassword) # Save changes to DB; done! Session.commit() return ""
def UserLogin(UserName, UserPassword, IgnoreRedirect = False): # Check for cookies - are we already logged in? if not IgnoreRedirect and session.get("UserName"): return "redirect" # Check for missing strings if not UserName: return "Error: Missing user name." if not UserPassword: return "Error: Missing password." # Does this username exist in the db? ExistingUsers = Session.query(UsersTable).filter(UsersTable.UserName == UserName).all() if len(ExistingUsers) <= 0: return "Error: User does not exist." # Pull up this user's information ExistingUser = ExistingUsers[0] # Hash password UserPassword = __GeneratePasswordHash(UserPassword) # Do the passwords match? if ExistingUser.UserPassword != UserPassword: return "Error: User password mismatch." # Update the login count by 1 and the login time ExistingUser.LogInCount += 1 ExistingUser.LastLogin = datetime.datetime.now() Session.commit() # Go ahead and start the cookie! session["UserID"] = ExistingUser.UserID session["UserName"] = ExistingUser.UserName session["UserPoints"] = UpdateUserPoints(ExistingUser.UserID) session["UserIconID"] = ExistingUser.IconID if ExistingUser.IsAdmin == 1: session["IsAdmin"] = True else: session["IsAdmin"] = False session.save() # All done return ""
def UserLogin(UserName, UserPassword, IgnoreRedirect=False): # Check for cookies - are we already logged in? if not IgnoreRedirect and session.get("UserName"): return "redirect" # Check for missing strings if not UserName: return "Error: Missing user name." if not UserPassword: return "Error: Missing password." # Does this username exist in the db? ExistingUsers = Session.query(UsersTable).filter(UsersTable.UserName == UserName).all() if len(ExistingUsers) <= 0: return "Error: User does not exist." # Pull up this user's information ExistingUser = ExistingUsers[0] # Hash password UserPassword = __GeneratePasswordHash(UserPassword) # Do the passwords match? if ExistingUser.UserPassword != UserPassword: return "Error: User password mismatch." # Update the login count by 1 and the login time ExistingUser.LogInCount += 1 ExistingUser.LastLogin = datetime.datetime.now() Session.commit() # Go ahead and start the cookie! session["UserID"] = ExistingUser.UserID session["UserName"] = ExistingUser.UserName session["UserPoints"] = UpdateUserPoints(ExistingUser.UserID) session["UserIconID"] = ExistingUser.IconID if ExistingUser.IsAdmin == 1: session["IsAdmin"] = True else: session["IsAdmin"] = False session.save() # All done return ""
def SetUserPassword(UserName, ExistingPassword, NewPassword, ConfirmPassword): # Is the form empty? Return no error... if not ExistingPassword and not NewPassword and not ConfirmPassword: return "" # Is the form partially filled? elif not ExistingPassword: return "Error: Missing existing password." elif not NewPassword: return "Error: Missing new password." elif not ConfirmPassword: return "Error: Missing confirmation password." # Get existing user ExistingUsers = Session.query(UsersTable).filter(UsersTable.UserName == UserName).all() if len(ExistingUsers) <= 0: return "Error: User name does not exist." ExistingUser = ExistingUsers[0] # Is the given password the active password? ExistingHash = __GeneratePasswordHash(ExistingPassword) if ExistingUser.UserPassword != ExistingHash: return "Error: Old password is not correct." # Check password lengths if len(NewPassword) <= 5: return "Error: New password is not long enough. Must be at least 6 characters long." if len(NewPassword) > 32: return "Error: New password is too long. May be at most 32 characters long." # Validate against whitelist to make sure these are valid characters if h.string_check_whitelist(NewPassword, h.string_whitelist_password) == False: return "Error: New password contains invalid characters." # Confirm the passwords are the same... if NewPassword != ConfirmPassword: return "Error: New passwords do not match." # All good to go - commit password changes ExistingUser.UserPassword = __GeneratePasswordHash(NewPassword) Session.commit() # Done! return "Success: Password has changed!"
def UserAddAchievement(UserID, AchievementID): # Does the current user have this achivement? UserAchievements = Achievements.AchievementsQueryUserID(UserID) for achievement in UserAchievements: if achievement.AchievementID == AchievementID: return # New achievement for this user! NewAchievement = UserAchievementsTable() NewAchievement.UserID = UserID NewAchievement.AchievementID = AchievementID # Commit to DB Session.add(NewAchievement) Session.commit() # Recompute the latest user points and force it onto this user UpdateUserPoints(UserID)
def UserReset(UserName): # Get existing user ExistingUsers = Session.query(UsersTable).filter(UsersTable.UserName == UserName).all() if len(ExistingUsers) <= 0: return "Error: User name does not exist." # Register this new user into the database ExistingUser = ExistingUsers[0] # Randomize an 8-character string NewPassword = "" for i in range(8): NewPassword += random.choice(h.string_whitelist) # Send an email with the password try: # Form a message message = """\ From: [email protected] To: %s Subject: Password Change New Password: \"%s\" (Change this as soon as possible!) Visit www.corecodex.com to learn more. """ % (ExistingUser.UserEMail, NewPassword) # Send mail s = smtplib.SMTP() s.connect() s.sendmail("*****@*****.**", ExistingUser.UserEMail, message) s.quit() except: return "Error: Unable to send email. Password was not reset." # Hash password ExistingUser.UserPassword = __GeneratePasswordHash(NewPassword) # Save changes to DB; done! Session.commit() return ""
def UserRegister(UserName, UserEMail, UserPassword, UserPasswordConfirm): # Check for cookies - are we already logged in? if session.get("UserName"): return "redirect" # Did we get the entire form correctly? if not UserName or not UserEMail or not UserPassword or not UserPasswordConfirm: return "Error: Missing fields. Please fill out all fields completely." # Check username lengths if len(UserName) <= 5: return "Error: User name is not long enough. Must be at least 6 characters long." if len(UserName) > 32: return "Error: User name is too long. May be at most 32 characters long." # Check email lengths if len(UserEMail) <= 5: return "Error: User e-mail is not long enough. Must be at least 6 characters long." if len(UserEMail) > 32: return "Error: User e-mail is too long. May be at most 32 characters long." # Check password lengths if len(UserPassword) <= 5: return "Error: User password is not long enough. Must be at least 6 characters long." if len(UserPassword) > 32: return "Error: User password is too long. May be at most 32 characters long." # Validate against whitelist to make sure these are valid characters if h.string_check_whitelist(UserName, h.string_whitelist) == False: return "Error: User name contains invalid characters." if h.string_check_whitelist(UserEMail, h.string_whitelist) == False: return "Error: User e-mail contains invalid characters." if h.string_check_whitelist(UserPassword, h.string_whitelist_password) == False: return "Error: Password contains invalid characters." # Confirm the passwords are the same... if UserPassword != UserPasswordConfirm: return "Error: Passwords do not match." # Finally, make sure no other users with the same name exist.. ExistingUsers = Session.query(UsersTable).filter(UsersTable.UserName == UserName).all() if len(ExistingUsers) > 0: return "Error: User name already exists! Please select a new user name." # Register this new user into the database NewUser = UsersTable() # Note that the user ID will auto-increment NewUser.UserName = UserName NewUser.UserEMail = UserEMail NewUser.UserPoints = 0 NewUser.UserPassword = __GeneratePasswordHash(UserPassword) NewUser.LogInCount = 0 NewUser.LastLogin = datetime.datetime.now() NewUser.IsAdmin = False NewUser.IconID = 0 # Commit to DB Session.add(NewUser) Session.commit() # Special achivement earned by people who register now - alpha testers UserAddAchievement(NewUser.UserID, 0) # All done return ""
def SolveChallenge(UserID, ChallengeID, ChallengeSourceCode, ChallengeLanguage): # Create a new solutions Solution = UserSolutionsTable() Solution.UserID = UserID Solution.ChallengeID = ChallengeID Solution.ResultCode = 0 Solution.ResultString = "Internal CoreJudge Failure." Solution.SampleResultString = "" # Leave empty Solution.MemoryUsage = 0 Solution.RuntimeUsage = 0 Solution.SourceCode = ChallengeSourceCode Solution.SourceLanguage = ChallengeLanguage Solution.SubmitDateTime = datetime.datetime.now() # Failure flag Failed = False # Load the challenge information Challenge = Challenges.ChallengeQueryID(ChallengeID) if not Challenge: Failed = True # Is this a supported language? # Formally accepted strings: Python, GCC C, C++, Java if not Failed: if ChallengeLanguage != "Python" and ChallengeLanguage != "GCC C" and ChallengeLanguage != "C++" and ChallengeLanguage != "Java": Failed = True # Check source code for viability to execute... if not Failed: SourceSafeErr = __IsSourceSafe(ChallengeSourceCode, ChallengeLanguage) if len(SourceSafeErr) > 0: Failed = True Solution.ResultCode = 8 # Attempt to execute the code against both the challenge test cases and the sample inputs if not Failed: # Execute the sample code Attempt = __ExecuteSource(Challenge.SampleCode, Challenge.SampleInput, Challenge.SampleOutput, ChallengeSourceCode, ChallengeLanguage, True) Solution.ResultString = Attempt.ResultString Solution.SampleResultString = Attempt.ResultString # Execute the official test case # Note that we don't post the result string, we just ignore the output # since we do the comparison of correct I/O within the exec source function Attempt = __ExecuteSource(Challenge.TestCode, Challenge.TestInput, Challenge.TestOutput, ChallengeSourceCode, ChallengeLanguage) Solution.MemoryUsage = Attempt.MemoryUsage Solution.RuntimeUsage = Attempt.RuntimeUsage Solution.ResultCode = Attempt.ResultCode Solution.ResultString = Attempt.ResultString # Commit to DB Session.add(Solution) Session.commit() # Force update the user's score if this solution is valid if Solution.ResultCode == 1: Users.UpdateUserPoints(UserID) # Return this new result ID (So the browser redirects) return Solution.SolutionID