mirror of
https://github.com/correl/mtgsqlive.git
synced 2024-12-01 11:09:57 +00:00
103 lines
No EOL
4.6 KiB
Python
Executable file
103 lines
No EOL
4.6 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
import json
|
|
import sqlite3
|
|
import time
|
|
import os
|
|
|
|
def getVal(data, field):
|
|
val = data.get(field)
|
|
if val:
|
|
return str(val)
|
|
return val
|
|
|
|
def fixJson(data):
|
|
if data:
|
|
return data.replace("'", '"')
|
|
return data
|
|
|
|
def create_db(database_connection):
|
|
c = database_connection.cursor()
|
|
c.execute('create table cards (id, layout, name, names, manaCost, cmc, colors, colorIdentity, type, supertypes, types, subtypes, rarity, text, flavor, artist, number, power, toughness, loyalty, multiverseid, variations, imageName, watermark, border, timeshifted, hand, life, reserved, releaseDate, starter, rulings, foreignNames, printings, originalText, originalType, legalities, source, setName, setCode, setReleaseDate)')
|
|
c.execute('create table lastUpdated (datetime)')
|
|
database_connection.commit()
|
|
c.close()
|
|
|
|
def json_to_db(json_file_opened, database_connection):
|
|
c = database_connection.cursor()
|
|
c.execute('insert into lastUpdated values (?)', [str(time.strftime("%Y-%m-%d %H:%M:%S"))])
|
|
|
|
# Get the setnames in the AllSets file and put them into a dictionary for later use
|
|
setNames = []
|
|
for i in json_file_opened.keys():
|
|
setNames.append(i)
|
|
|
|
# Iterate through each set, one at a time
|
|
for thisSet in setNames:
|
|
data = json_file_opened[thisSet] # All of the data for the set (I.e. SOI-x.json)
|
|
|
|
setName = data["name"]
|
|
setReleaseDate = data["releaseDate"]
|
|
|
|
setCards = data["cards"] # Now iterate through the setCards for each card in the set
|
|
|
|
for thisCard in setCards:
|
|
thisCard_id = getVal(thisCard, "id")
|
|
layout = getVal(thisCard, "layout")
|
|
name = getVal(thisCard, "name")
|
|
names = getVal(thisCard, "names")
|
|
manaCost = getVal(thisCard, "manaCost")
|
|
cmc = getVal(thisCard, "cmc")
|
|
colors = fixJson( getVal(thisCard, "colors") )
|
|
colorIdentity = fixJson( getVal(thisCard, "colorIdentity") )
|
|
thisCard_type = getVal(thisCard, "type")
|
|
supertypes = fixJson( getVal(thisCard, "supertypes") )
|
|
types = fixJson( getVal(thisCard, "types") )
|
|
subtypes = fixJson( getVal(thisCard, "subtypes") )
|
|
rarity = getVal(thisCard, "rarity")
|
|
text = getVal(thisCard, "text")
|
|
flavor = getVal(thisCard, "flavor")
|
|
artist = getVal(thisCard, "artist")
|
|
number = getVal(thisCard, "number")
|
|
power = getVal(thisCard, "power")
|
|
toughness = getVal(thisCard, "toughness")
|
|
loyalty = getVal(thisCard, "loyalty")
|
|
multiverseid = getVal(thisCard, "multiverseid")
|
|
variations = getVal(thisCard, "variations")
|
|
imageName = getVal(thisCard, "imageName")
|
|
watermark = getVal(thisCard, "watermark")
|
|
border = getVal(thisCard, "border")
|
|
timeshifted = getVal(thisCard, "timeshifted")
|
|
hand = getVal(thisCard, "hand")
|
|
life = getVal(thisCard, "life")
|
|
reserved = getVal(thisCard, "reserved")
|
|
releaseDate = getVal(thisCard, "releaseDate")
|
|
starter = getVal(thisCard, "starter")
|
|
rulings = fixJson( getVal(thisCard, "rulings") )
|
|
foreignNames = fixJson( getVal(thisCard, "foreignNames") )
|
|
printings = fixJson( getVal(thisCard, "printings") )
|
|
originalText = getVal(thisCard, "originalText")
|
|
originalType = getVal(thisCard, "originalType")
|
|
legalities = fixJson( getVal(thisCard, "legalities") )
|
|
source = getVal(thisCard, "source")
|
|
|
|
thisCard_data = [thisCard_id, layout, name, names, manaCost, cmc, colors, colorIdentity, thisCard_type, supertypes, types, subtypes, rarity, text, flavor, artist, number, power, toughness, loyalty, multiverseid, variations, imageName, watermark, border, timeshifted, hand, life, reserved, releaseDate, starter, rulings, foreignNames, printings, originalText, originalType, legalities, source, setName, thisSet, setReleaseDate]
|
|
|
|
c.execute('insert into cards values (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)', thisCard_data)
|
|
database_connection.commit()
|
|
c.close()
|
|
|
|
def main():
|
|
i = input("Create new database? 1 or 0: ")
|
|
d = os.path.join(os.path.expanduser(input("Location of database: ")), "Magic DB.db");
|
|
d = sqlite3.connect(d)
|
|
|
|
if (i == '1'):
|
|
create_db(d)
|
|
|
|
xml = os.path.join(os.path.expanduser(input("Location of AllSets-x.json: ")), "AllSets-x.json")
|
|
xml = json.load(open(xml, 'r'))
|
|
|
|
json_to_db(xml, d)
|
|
|
|
if __name__ == '__main__':
|
|
main() |