Import project
This commit is contained in:
67
main.py
Normal file
67
main.py
Normal file
@@ -0,0 +1,67 @@
|
||||
import discogs_client, json, re, urllib.request, os
|
||||
|
||||
client = discogs_client.Client('ExampleApplication/0.1', user_token="")
|
||||
|
||||
band_id =
|
||||
member_id =
|
||||
discogs_id = input("Enter Discogs ID: ")
|
||||
band = client.artist(discogs_id)
|
||||
|
||||
band_json = {}
|
||||
band_json['name'] = band.name
|
||||
band_json['id'] = band_id
|
||||
band_json['description'] = band.profile
|
||||
band_json['image'] = band.images[0]['uri']
|
||||
band_json['enabled'] = 0
|
||||
|
||||
members_json = []
|
||||
|
||||
def resolve_ref(match):
|
||||
print(f"Resolving {match.group()} => ", end="")
|
||||
ref = client.artist(int(match.group()[2:-1]))
|
||||
print(f"{ref.name}")
|
||||
return ref.name
|
||||
|
||||
def sanitize_path(path):
|
||||
return path.replace(" ", "_")
|
||||
|
||||
def image_closest_to_aspect(images):
|
||||
aspect = 0.6
|
||||
closest = None
|
||||
for image in images:
|
||||
if image['width'] / image['height'] > aspect and image['width'] < image['height']:
|
||||
if closest is None or image['width'] < closest['width']:
|
||||
closest = image
|
||||
return closest if closest is not None else images[0]
|
||||
|
||||
band_folder = sanitize_path(band.name)
|
||||
if not os.path.exists(f"assets/{band_folder}"):
|
||||
os.makedirs(f"assets/{band_folder}")
|
||||
|
||||
|
||||
for member in band.members:
|
||||
print(member.name)
|
||||
profile = str(member.profile)
|
||||
profile = re.sub(r'\[a[0-9|a-zA-Z]*\]', resolve_ref, profile)
|
||||
profile = re.sub(r'\[a=[0-9|a-zA-Z ]*\]', lambda match: match.group()[3:-1], profile)
|
||||
|
||||
print("Downloading image...")
|
||||
image_path = f"{band_folder}/{member.id}_{sanitize_path(member.name)}.jpeg"
|
||||
urllib.request.urlretrieve(image_closest_to_aspect(member.images)['uri'], f"assets/{image_path}")
|
||||
|
||||
member = {
|
||||
'id': member_id,
|
||||
'bandId': band_id,
|
||||
'name': member.name,
|
||||
'description': profile,
|
||||
'imageIdentifier': image_path,
|
||||
'enabled': 0
|
||||
}
|
||||
members_json.append(member)
|
||||
member_id += 1
|
||||
|
||||
with open(f"aseets/import/characters/{band_folder}_characters.json", "w") as characters_file:
|
||||
json.dump(members_json, characters_file, indent=4, sort_keys=False)
|
||||
|
||||
with open(f"assets/import/bands/{band_folder}.json", "w") as band_file:
|
||||
json.dump(band_json, band_file, indent=4, sort_keys=False)
|
||||
Reference in New Issue
Block a user