lrrbot/build_carddb_altered.py at master · mrphlip/lrrbot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
import json
import psycopg2

from common.card import clean_text, CARD_GAME_ALTERED
from common.config import config

sets = {
	# Beyond the Gates - KS Edition
	'COREKS': {
		'release_date': '2024-07-15',
	},
	# Beyond the Gates
	'CORE': {
		'release_date': '2024-09-13',
	},
	# Trial by Frost
	'ALIZE': {
		'release_date': '2025-01-31',
	},
}

def build_card_description(name, card):
	yield name
	if card['cardType']['reference'] != 'HERO' and card['cardType']['reference'] != 'TOKEN':
		yield ' [hand: '
		yield card['elements']['MAIN_COST'].strip('#')
		yield ', reserve: '
		yield card['elements']['RECALL_COST'].strip('#')
		yield ']'
	yield ' | '
	if card['cardType']['reference'] == 'HERO':
		yield card['mainFaction']['name']
		yield ' '
	yield card['cardType']['name']
	if card['cardSubTypes']:
		yield ' \u2014 '
		for i, subtype in enumerate(card['cardSubTypes']):
			if i != 0:
				yield ', '
			yield subtype['name']

	reserve = card['elements'].get('RESERVE', '').strip("#")
	landmarks = card['elements'].get('PERMANENT', '').strip("#")
	if reserve or landmarks:
		yield ' [reserve: '
		yield reserve or '?'
		yield ', landmarks: '
		yield landmarks or '?'
		yield ']'

	forest_power = card['elements'].get('FOREST_POWER', '').strip('#')
	mountain_power = card['elements'].get('MOUNTAIN_POWER', '').strip('#')
	ocean_power = card['elements'].get('OCEAN_POWER', '').strip('#')
	if forest_power or mountain_power or ocean_power:
		yield ' ['
		yield forest_power or '?'
		yield '/'
		yield mountain_power or '?'
		yield '/'
		yield ocean_power or '?'
		yield ']'

	main_ability = card['elements'].get('MAIN_EFFECT', '').replace('#', '').replace('\u00A0', ' ')
	support_ability = card['elements'].get('ECHO_EFFECT', '').replace('#', '').replace('\u00A0', ' ')
	if main_ability or support_ability:
		yield ' | '
		if main_ability:
			yield main_ability
		if main_ability and support_ability:
			yield ' / '
		if support_ability:
			yield support_ability

def build_card_name(card):
	if '-C-' in card['collectorNumberFormatted']:
		return f"{card['name']} (Common)"
	if '-R-' in card['collectorNumberFormatted']:
		return f"{card['name']} (Rare)"
	if '-F-' in card['collectorNumberFormatted']:
		return f"{card['name']} (Faction-shifted)"
	return card['name']

if __name__ == '__main__':
	print("Loading cards...")

	with open("carddb-altered.json") as f:
		cards = json.load(f)

	processed = {}
	codes = {}

	print("Processing cards...")

	for card in cards:
		if card['cardProduct']['reference'] == 'P':
			# Promo cards have broken card entries
			print(f"WARNING: Ignoring {card['cardSet']['name']} promo card {card['name']}")
			continue
		name = build_card_name(card)
		filtered_name = clean_text(name)
		release_date = sets[card['cardSet']['reference']]['release_date']
		codes.setdefault(filtered_name, []).append(card['reference'])
		if filtered_name not in processed or processed[filtered_name]['release_date'] < release_date:
			processed[filtered_name] = {
				'name': name,
				'text': ''.join(build_card_description(name, card)),
				'release_date': release_date,
			}

	print("Updating card database...")

	with psycopg2.connect(config['postgres']) as conn, conn.cursor() as cur:
		cur.execute("DELETE FROM cards WHERE game = %s", (CARD_GAME_ALTERED, ))
		for filtered_name, card in processed.items():
			cur.execute(
				"INSERT INTO cards (game, filteredname, name, text) VALUES (%s, %s, %s, %s) RETURNING id",
				(
					CARD_GAME_ALTERED,
					filtered_name,
					card['name'],
					card['text'],
				)
			)
			card_id, = cur.fetchone()
			for code in codes.get(filtered_name, []):
				cur.execute(
					"INSERT INTO card_codes (code, cardid, game) VALUES (%s, %s, %s)",
					(
						code,
						card_id,
						CARD_GAME_ALTERED,
					)
				)

	print("Done.")