mirror of
https://github.com/ldericher/fftcgtool
synced 2025-01-15 15:02:59 +00:00
Begin rewrite
This commit is contained in:
parent
cc5955865b
commit
cd115e2f9e
11 changed files with 124 additions and 257 deletions
3
.idea/.gitignore
vendored
Normal file
3
.idea/.gitignore
vendored
Normal file
|
|
@ -0,0 +1,3 @@
|
|||
# Default ignored files
|
||||
/shelf/
|
||||
/workspace.xml
|
||||
8
.idea/fftcgtool_v2.iml
Normal file
8
.idea/fftcgtool_v2.iml
Normal file
|
|
@ -0,0 +1,8 @@
|
|||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<module type="PYTHON_MODULE" version="4">
|
||||
<component name="NewModuleRootManager">
|
||||
<content url="file://$MODULE_DIR$" />
|
||||
<orderEntry type="inheritedJdk" />
|
||||
<orderEntry type="sourceFolder" forTests="false" />
|
||||
</component>
|
||||
</module>
|
||||
6
.idea/inspectionProfiles/profiles_settings.xml
Normal file
6
.idea/inspectionProfiles/profiles_settings.xml
Normal file
|
|
@ -0,0 +1,6 @@
|
|||
<component name="InspectionProjectProfileManager">
|
||||
<settings>
|
||||
<option name="USE_PROJECT_PROFILE" value="false" />
|
||||
<version value="1.0" />
|
||||
</settings>
|
||||
</component>
|
||||
4
.idea/misc.xml
Normal file
4
.idea/misc.xml
Normal file
|
|
@ -0,0 +1,4 @@
|
|||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<project version="4">
|
||||
<component name="ProjectRootManager" version="2" project-jdk-name="Pipenv (fftcgtool_v2)" project-jdk-type="Python SDK" />
|
||||
</project>
|
||||
8
.idea/modules.xml
Normal file
8
.idea/modules.xml
Normal file
|
|
@ -0,0 +1,8 @@
|
|||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<project version="4">
|
||||
<component name="ProjectModuleManager">
|
||||
<modules>
|
||||
<module fileurl="file://$PROJECT_DIR$/.idea/fftcgtool_v2.iml" filepath="$PROJECT_DIR$/.idea/fftcgtool_v2.iml" />
|
||||
</modules>
|
||||
</component>
|
||||
</project>
|
||||
12
Pipfile
Normal file
12
Pipfile
Normal file
|
|
@ -0,0 +1,12 @@
|
|||
[[source]]
|
||||
name = "pypi"
|
||||
url = "https://pypi.org/simple"
|
||||
verify_ssl = true
|
||||
|
||||
[dev-packages]
|
||||
|
||||
[packages]
|
||||
requests = "*"
|
||||
|
||||
[requires]
|
||||
python_version = "3.6"
|
||||
60
Pipfile.lock
generated
Normal file
60
Pipfile.lock
generated
Normal file
|
|
@ -0,0 +1,60 @@
|
|||
{
|
||||
"_meta": {
|
||||
"hash": {
|
||||
"sha256": "8739d581819011fea34feca8cc077062d6bdfee39c7b37a8ed48c5e0a8b14837"
|
||||
},
|
||||
"pipfile-spec": 6,
|
||||
"requires": {
|
||||
"python_version": "3.6"
|
||||
},
|
||||
"sources": [
|
||||
{
|
||||
"name": "pypi",
|
||||
"url": "https://pypi.org/simple",
|
||||
"verify_ssl": true
|
||||
}
|
||||
]
|
||||
},
|
||||
"default": {
|
||||
"certifi": {
|
||||
"hashes": [
|
||||
"sha256:2bbf76fd432960138b3ef6dda3dde0544f27cbf8546c458e60baf371917ba9ee",
|
||||
"sha256:50b1e4f8446b06f41be7dd6338db18e0990601dce795c2b1686458aa7e8fa7d8"
|
||||
],
|
||||
"version": "==2021.5.30"
|
||||
},
|
||||
"charset-normalizer": {
|
||||
"hashes": [
|
||||
"sha256:0c8911edd15d19223366a194a513099a302055a962bca2cec0f54b8b63175d8b",
|
||||
"sha256:f23667ebe1084be45f6ae0538e4a5a865206544097e4e8bbcacf42cd02a348f3"
|
||||
],
|
||||
"markers": "python_version >= '3'",
|
||||
"version": "==2.0.4"
|
||||
},
|
||||
"idna": {
|
||||
"hashes": [
|
||||
"sha256:14475042e284991034cb48e06f6851428fb14c4dc953acd9be9a5e95c7b6dd7a",
|
||||
"sha256:467fbad99067910785144ce333826c71fb0e63a425657295239737f7ecd125f3"
|
||||
],
|
||||
"markers": "python_version >= '3'",
|
||||
"version": "==3.2"
|
||||
},
|
||||
"requests": {
|
||||
"hashes": [
|
||||
"sha256:6c1246513ecd5ecd4528a0906f910e8f0f9c6b8ec72030dc9fd154dc1a6efd24",
|
||||
"sha256:b8aa58f8cf793ffd8782d3d8cb19e66ef36f7aba4353eec859e74678b01b07a7"
|
||||
],
|
||||
"index": "pypi",
|
||||
"version": "==2.26.0"
|
||||
},
|
||||
"urllib3": {
|
||||
"hashes": [
|
||||
"sha256:39fb8672126159acb139a7718dd10806104dec1e2f0f6c88aab05d17df10c8d4",
|
||||
"sha256:f57b4c16c62fa2760b7e3d97c35b255512fb6b59a259730f36ba32ce9f8e342f"
|
||||
],
|
||||
"markers": "python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3, 3.4' and python_version < '4'",
|
||||
"version": "==1.26.6"
|
||||
}
|
||||
},
|
||||
"develop": {}
|
||||
}
|
||||
89
card.py
89
card.py
|
|
@ -1,89 +0,0 @@
|
|||
import requests
|
||||
|
||||
from PIL import Image
|
||||
from io import BytesIO
|
||||
|
||||
# Card front by Square API
|
||||
FACEURL = "https://fftcg.cdn.sewest.net/images/cards/full/{}_eg.jpg"
|
||||
|
||||
# Card back image by Aurik
|
||||
BACKURL = "http://cloud-3.steamusercontent.com/ugc/948455238665576576/85063172B8C340602E8D6C783A457122F53F7843/"
|
||||
|
||||
class Card:
|
||||
def __init__(self, data):
|
||||
# check if this is a card back
|
||||
if data == 0:
|
||||
self._serial = "0-000"
|
||||
self._name = "[cardback]"
|
||||
self._rarity = "X"
|
||||
self._element = "None"
|
||||
self._elements = None
|
||||
|
||||
self._description = "None"
|
||||
self._iurl = BACKURL
|
||||
|
||||
# else import from data
|
||||
else:
|
||||
self._serial = data["serial_number"]
|
||||
self._name = data["name"]
|
||||
self._rarity = data["rarity"][0]
|
||||
self._element = data["elements"][0]
|
||||
|
||||
# handle multi element cards
|
||||
if len(data["elements"]) > 1:
|
||||
self._element = "Multi"
|
||||
self._elements = data["elements"]
|
||||
|
||||
# fix db typos
|
||||
if self._serial == "12-052":
|
||||
self._rarity = "H"
|
||||
elif self._serial == "12-049":
|
||||
self._rarity = "H"
|
||||
|
||||
self._description = "\n\n".join(data["abilities"])
|
||||
self._iurl = FACEURL.format(self.get_id()) # official url
|
||||
#self._iurl = data["image"] # ffdecks url
|
||||
|
||||
# 'Shinra' (Wind, 6-048C)
|
||||
def __str__(self):
|
||||
return "'{}' ({}, {})".format(self._name, self._element, self.get_id())
|
||||
|
||||
# 6-048C
|
||||
def get_id(self):
|
||||
rarity = self._rarity if self._rarity in ["C", "R", "H", "L", "S"] else ""
|
||||
return "{}{}".format(self._serial, rarity)
|
||||
|
||||
# return in dictionary format
|
||||
def get_dict(self):
|
||||
return {
|
||||
"Nickname": self._name,
|
||||
"Description": self._description,
|
||||
|
||||
"Name": "Card",
|
||||
"Transform": {
|
||||
"scaleX": 2.17822933,
|
||||
"scaleY": 1.0,
|
||||
"scaleZ": 2.17822933
|
||||
},
|
||||
"Locked": False,
|
||||
"Grid": True,
|
||||
"Snap": True,
|
||||
"Autoraise": True,
|
||||
"Sticky": True,
|
||||
"Tooltip": True,
|
||||
"GridProjection": False,
|
||||
"SidewaysCard": False,
|
||||
"Hands": True
|
||||
}
|
||||
|
||||
# download and resize card image
|
||||
def get_image(self, resolution):
|
||||
try:
|
||||
response = requests.get(self._iurl)
|
||||
except:
|
||||
return False
|
||||
|
||||
im = Image.open(BytesIO(response.content))
|
||||
im = im.convert("RGB")
|
||||
im = im.resize(resolution, Image.BICUBIC)
|
||||
return im
|
||||
0
fftcg/__init__.py
Normal file
0
fftcg/__init__.py
Normal file
23
fftcg/card.py
Normal file
23
fftcg/card.py
Normal file
|
|
@ -0,0 +1,23 @@
|
|||
import re
|
||||
|
||||
|
||||
class Card:
|
||||
def __init__(self, data):
|
||||
code_match = re.match(r'([0-9]+)-([0-9]+)([CRHLSB])', data["Code"])
|
||||
if code_match:
|
||||
self._opus, self._serial, self._rarity = code_match.groups()
|
||||
|
||||
else:
|
||||
self._opus = "PR"
|
||||
self._serial = re.match(r'PR-([0-9]+)', data["Code"]).group(1)
|
||||
self._rarity = "P"
|
||||
|
||||
self._name = data["Name_EN"]
|
||||
self._element = data["Element"]
|
||||
|
||||
def __str__(self):
|
||||
return f"'{self._name}' ({self._element}, {self.get_id()})"
|
||||
|
||||
# 6-048C
|
||||
def get_id(self):
|
||||
return f"{self._opus}-{self._serial}{self._rarity}"
|
||||
168
opus.py
168
opus.py
|
|
@ -1,168 +0,0 @@
|
|||
import queue
|
||||
import threading
|
||||
import logging
|
||||
import json
|
||||
|
||||
from card import Card, BACKURL
|
||||
from PIL import Image
|
||||
|
||||
|
||||
# multithreaded card image loading
|
||||
class imageLoader(threading.Thread):
|
||||
def __init__(self, queue, composite, composite_lock, grid, resolution):
|
||||
threading.Thread.__init__(self)
|
||||
self.__queue = queue
|
||||
self.__composite = composite
|
||||
self.__lock = composite_lock
|
||||
self.__grid = grid
|
||||
self.__resolution = resolution
|
||||
|
||||
def run(self):
|
||||
logger = logging.getLogger(__name__)
|
||||
# shorthands
|
||||
r, c = self.__grid # rows and columns per sheet
|
||||
w, h = self.__resolution # width, height per card
|
||||
|
||||
while not self.__queue.empty():
|
||||
# take next card
|
||||
i, card = self.__queue.get()
|
||||
|
||||
# fetch card image (retry on fail)
|
||||
while True:
|
||||
logger.info("get image for card {}".format(card))
|
||||
im = card.get_image(self.__resolution)
|
||||
if im:
|
||||
break
|
||||
|
||||
# paste image in correct position
|
||||
self.__lock.acquire()
|
||||
x, y = (i % c) * w, (i // c) * h
|
||||
logger.info("paste image {} at P{}({}, {})".format(im.mode, i, x, y))
|
||||
self.__composite.paste(im, (x, y, x+w, y+h))
|
||||
self.__lock.release()
|
||||
|
||||
# image is processed
|
||||
self.__queue.task_done()
|
||||
|
||||
|
||||
class Opus:
|
||||
def __init__(self, data):
|
||||
logger = logging.getLogger(__name__)
|
||||
self._cards = []
|
||||
|
||||
for card_data in data:
|
||||
card = Card(card_data)
|
||||
logger.info("Imported card {}".format(card))
|
||||
self._cards.append(card)
|
||||
|
||||
# sort every element alphabetically
|
||||
self._cards.sort(key=lambda x: x._serial)
|
||||
self._cards.sort(key=lambda x: x._name)
|
||||
self._cards.sort(key=lambda x: x._element)
|
||||
|
||||
def __get_sheets(self, grid):
|
||||
# cards per sheet
|
||||
count = grid[0]*grid[1] - 1
|
||||
# flat copy
|
||||
cards = self._cards
|
||||
|
||||
# while there are cards
|
||||
while cards:
|
||||
# get a chunk
|
||||
yield cards[:count]
|
||||
# remove that chunk
|
||||
cards = cards[count:]
|
||||
|
||||
def get_images(self, grid, resolution, threadnum=16):
|
||||
logger = logging.getLogger(__name__)
|
||||
# shorthands
|
||||
r, c = grid # rows and columns per sheet
|
||||
w, h = resolution # width, height per card
|
||||
|
||||
for sheet in self.__get_sheets(grid):
|
||||
# enqueue a sheet of cards
|
||||
cardQueue = queue.Queue()
|
||||
for i, card in enumerate(sheet):
|
||||
cardQueue.put((i, card))
|
||||
|
||||
# add hidden face
|
||||
hidden = Card(0)
|
||||
cardQueue.put((r * c - 1, hidden))
|
||||
|
||||
# create a new card sheet
|
||||
sheet = Image.new("RGB", (c*w, r*h))
|
||||
logger.info("New image: {}x{}".format(*sheet.size))
|
||||
|
||||
# beware concurrent paste
|
||||
sheet_lock = threading.Lock()
|
||||
|
||||
# start multithreading, wait for finish
|
||||
for _ in range(threadnum):
|
||||
imageLoader(cardQueue, sheet, sheet_lock, grid, resolution).start()
|
||||
cardQueue.join()
|
||||
|
||||
# sheet image is generated, return now
|
||||
yield sheet
|
||||
|
||||
def get_json(self, opusid, deckname, grid, cardfilter, faceurls):
|
||||
# shorthands
|
||||
r, c = grid # rows and columns per sheet
|
||||
|
||||
jsondict = { "ObjectStates": [ {
|
||||
"Name": "Deck",
|
||||
"Nickname": "Opus {} {}".format(opusid, deckname),
|
||||
"Description": "",
|
||||
|
||||
"Transform": {
|
||||
"scaleX": 2.17822933,
|
||||
"scaleY": 1.0,
|
||||
"scaleZ": 2.17822933
|
||||
},
|
||||
|
||||
"Locked": False,
|
||||
"Grid": True,
|
||||
"Snap": True,
|
||||
"Autoraise": True,
|
||||
"Sticky": True,
|
||||
"Tooltip": True,
|
||||
"GridProjection": False,
|
||||
"Hands": False,
|
||||
"SidewaysCard": False,
|
||||
|
||||
"DeckIDs": [],
|
||||
"CustomDeck": {},
|
||||
"ContainedObjects": []
|
||||
} ] }
|
||||
|
||||
for sheetnum, sheet in enumerate(self.__get_sheets(grid)):
|
||||
# get current face
|
||||
faceurl = faceurls[sheetnum]
|
||||
# first sheet is "CustomDeck 1"
|
||||
sheetnum = sheetnum + 1
|
||||
|
||||
# recurring sheet dictionary
|
||||
sheetdict = { str(sheetnum): {
|
||||
"FaceURL": faceurl,
|
||||
"BackURL": BACKURL,
|
||||
"NumWidth": c,
|
||||
"NumHeight": r
|
||||
} }
|
||||
|
||||
for cardnum, card in enumerate(sheet):
|
||||
if not cardfilter(card):
|
||||
continue
|
||||
|
||||
# cardid 123 is on "CustomDeck 1", 3rd row, 4th column
|
||||
cardid = sheetnum * 100 + (cardnum // c) * 10 + (cardnum % c) * 1
|
||||
jsondict["ObjectStates"][0]["DeckIDs"].append(cardid)
|
||||
|
||||
# Add card object to ContainedObjects
|
||||
carddict = card.get_dict()
|
||||
carddict["CardID"] = cardid
|
||||
carddict["CustomDeck"] = sheetdict
|
||||
jsondict["ObjectStates"][0]["ContainedObjects"].append(carddict)
|
||||
|
||||
# Add sheet to CustomDecks
|
||||
jsondict["ObjectStates"][0]["CustomDeck"].update(sheetdict)
|
||||
|
||||
return json.dumps(jsondict, indent=2)
|
||||
Loading…
Reference in a new issue