mirror of
https://github.com/ParadiseSS13/Paradise.git
synced 2025-12-19 14:51:27 +00:00
95ce462b14
* Refactors Karma * I FELL VICTIM TO MY OWN CI * Update code/modules/client/client_procs.dm * Update SQL/updates/31-32.py * Mochi tweaks Co-authored-by: dearmochi <1496804+dearmochi@users.noreply.github.com> Co-authored-by: dearmochi <1496804+dearmochi@users.noreply.github.com>
99 lines
2.9 KiB
Python
99 lines
2.9 KiB
Python
# Listen up.
|
|
# In order to run this script on Windows, you need to make sure you have Python **3** installed. Tested on 3.8.2
|
|
# It won't work on 2.7 at all.
|
|
|
|
# To run this, supply the following args in a command shell
|
|
# python 31-32.py address username password database
|
|
# Example:
|
|
# python 31-32.py 172.16.0.200 affectedarc07 (redacted) paradise_gamedb
|
|
|
|
#!/usr/bin/env python3
|
|
import mysql.connector, argparse
|
|
from mysql.connector.errors import IntegrityError
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("address", help="MySQL server address (use localhost for the current computer)")
|
|
parser.add_argument("username", help="MySQL login username")
|
|
parser.add_argument("password", help="MySQL login password")
|
|
parser.add_argument("database", help="Database name")
|
|
|
|
args = parser.parse_args()
|
|
db = mysql.connector.connect(host=args.address, user=args.username, passwd=args.password, db=args.database)
|
|
cursor = db.cursor()
|
|
|
|
# Dictionaries of old & new purchase IDs
|
|
job_purchase_map = {
|
|
"Blueshield": "JOB_Blueshield",
|
|
"Barber": "JOB_Barber",
|
|
"Brig Physician": "JOB_BrigPhysician",
|
|
"Nanotrasen Representative": "JOB_NanotrasenRepresentative",
|
|
"Security Pod Pilot": "JOB_SecurityPodPilot",
|
|
"Mechanic": "JOB_Mechanic",
|
|
"Magistrate": "JOB_Magistrate"
|
|
}
|
|
|
|
species_purchase_map = {
|
|
"Grey": "SPECIES_Grey",
|
|
"Kidan": "SPECIES_Kidan",
|
|
"Slime People": "SPECIES_SlimePeople",
|
|
"Vox": "SPECIES_Vox",
|
|
"Drask": "SPECIES_Drask",
|
|
"Machine": "SPECIES_Machine",
|
|
"Plasmaman": "SPECIES_Plasmaman"
|
|
}
|
|
|
|
print("Connected")
|
|
|
|
cursor.execute("SELECT ckey, job, species FROM whitelist")
|
|
data = cursor.fetchall()
|
|
|
|
print("Loaded {} rows from old table".format(len(data)))
|
|
|
|
new_rows = []
|
|
|
|
for entry in data:
|
|
job_set = entry[1]
|
|
species_set = entry[2]
|
|
|
|
# Account for null entries
|
|
if job_set is not None:
|
|
jobs = job_set.split(",")
|
|
else:
|
|
jobs = []
|
|
|
|
if species_set is not None:
|
|
species = species_set.split(",")
|
|
else:
|
|
species = []
|
|
|
|
if len(jobs) > 0:
|
|
for job in jobs:
|
|
if job in job_purchase_map:
|
|
new_job = job_purchase_map[job]
|
|
new_row = [entry[0], new_job]
|
|
new_rows.append(new_row)
|
|
|
|
if len(species) > 0:
|
|
for race in species:
|
|
if race in species_purchase_map:
|
|
new_species = species_purchase_map[race]
|
|
new_row = [entry[0], new_species]
|
|
new_rows.append(new_row)
|
|
|
|
print("Generated {} rows to insert into new table".format(len(new_rows)))
|
|
|
|
stmt = "INSERT INTO karma_purchases (ckey, purchase) VALUES (%s, %s)"
|
|
|
|
print("Inserting...")
|
|
for row in new_rows:
|
|
data = (row[0], row[1])
|
|
try:
|
|
cursor.execute(stmt, data)
|
|
except IntegrityError:
|
|
print("{} already had {} as a purchase. They may be eligble for compensation".format(row[0], row[1])) # They had an accident that wasnt their fault
|
|
|
|
cursor.close()
|
|
print("Saving...")
|
|
db.commit()
|
|
print("Done!")
|