135 lines
3.4 KiB

import json
import requests
import sys
import mysql.connector
import datetime
import time
from typing import List
class Application:
def __init__(self, username):
Class containing application data
:param username: player's username
self.username = username
def to_insert_query(self):
Created an SQL insert query from the application
:return: SQL query string
return """\
INSERT INTO minecraft_manager_application
(username, age, player_type, ever_banned, read_rules, accepted, date)
("{}", 16, "Imported from Lasagn Whitelist", FALSE, "Yes", TRUE, "{}");
""".format(self.username, datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
def __str__(self):
return self.username
def get_latest_username(uuid: str) -> str:
Gets the latest username of a player from the mojang api
:param uuid: player's uuid
:return: player name
r = requests.get("https://api.mojang.com/user/profiles/{}/names".format(uuid))
if r.status_code == 200:
j = r.json()
return j[-1]["name"]
raise Exception("Error getting user {}. Code: {}: {}".format(uuid, r.status_code, r.text))
def parse_whitelist(whitelist_path: str) -> List[Application]:
Process all the applications from the whitelist file
:return: list of applications
# Load json from file
with open(whitelist_path, "r") as f:
whitelist = json.load(f)
# Parse applications from whitelist data
applications = []
for entry in whitelist:
uuid = entry["uuid"]
username = get_latest_username(uuid)
app = Application(username)
# Prevent Mojang from rate limiting us
return applications
def insert_apps_into_db(apps: List[Application], host: str, user: str, pw: str, mcm_db_name: str):
Insert applications into the MCM database
:param apps: list of applications to import
:param host: mysql host
:param user: mysql user
:param pw: mysql user password
:param mcm_db_name: MCM database name
# Open database
db = mysql.connector.connect(host=host, user=user, password=pw, database=mcm_db_name)
cursor = db.cursor(buffered=True)
for app in apps:
# Check if the player already has an application
cursor.execute("SELECT * FROM minecraft_manager_application WHERE username = '{}';".format(app.username))
result = cursor.fetchall()
# If there is not an existing app, insert the application into the DB
if len(result) == 0:
# Commit and close up
def main():
if len(sys.argv) < 6:
print("{} [WHITELIST PATH] [MYSQL HOST] [MYSQL USER] [MYSQL PW] [MCM DB Name]".format(sys.argv[0]))
# Parse args
whitelist_path = sys.argv[1]
host = sys.argv[2]
user = sys.argv[3]
pw = sys.argv[4]
mc_db_name = sys.argv[5]
# Parse whitelist
apps = parse_whitelist(whitelist_path)
# Insert apps into the database
insert_apps_into_db(apps, host, user, pw, mc_db_name)
if __name__ == '__main__':