Skip to content

Commit d4ba90f

Browse files
committed
add just a card
you can add a lot of them, it should be light on tombll.db file, but you can use -u flag after
1 parent 1cbbe0e commit d4ba90f

File tree

2 files changed

+31
-1
lines changed

2 files changed

+31
-1
lines changed

database/scrape_trle.py

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -299,6 +299,23 @@ def get_trle_walkthrough(level_soup):
299299
return None
300300

301301

302+
def get_trle_level_card(soup, data):
303+
"""Call all the other soup extracts for TRLE card."""
304+
data['title'] = scrape_common.get_trle_title(soup)
305+
data['authors'] = scrape_common.get_trle_authors(soup)
306+
data['type'] = scrape_common.get_trle_type(soup)
307+
data['class'] = scrape_common.get_trle_class(soup)
308+
data['release'] = scrape_common.get_trle_release(soup)
309+
data['difficulty'] = scrape_common.get_trle_difficulty(soup)
310+
data['duration'] = scrape_common.get_trle_duration(soup)
311+
data['screen'] = scrape_common.get_trle_screen(soup)
312+
data['large_screens'] = []
313+
data['trle_id'] = scrape_common.trle_url_to_int(soup.find('a', string='Download').get('href'))
314+
data['zip_files'] = scrape_trle_download.get_zip_file_info(data['trle_id'])
315+
data['body'] = ""
316+
data['walkthrough'] = ""
317+
318+
302319
def get_trle_level(soup, data):
303320
"""Call all the other soup extracts for TRLE."""
304321
data['title'] = scrape_common.get_trle_title(soup)
@@ -336,7 +353,7 @@ def get_trle_level(soup, data):
336353

337354
def get_trle_index(title):
338355
"""
339-
Search for a level by its title and returns the URLs.
356+
Search for a level by its title and returns the URLs.
340357
341358
Parameters:
342359
title (str): The title of the level to search for.

database/tombll_manage_data.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ def print_info():
2828
-a [lid] Add a level record
2929
-aj [lid path] Download a level record to json file
3030
-af [path] Add from the json file
31+
-ac [lid] Add a level card record without info and walkthrough
3132
-rm [lid] Remove one level
3233
-u [lid] Update a level record
3334
@@ -330,6 +331,18 @@ def update_tombll_json_to_database(data, level_id, con):
330331
main_con.commit()
331332
print(f"File {sys.argv[2]} added successfully.")
332333

334+
elif (sys.argv[1] == "-ac" and number_of_argument == 3):
335+
main_cur = main_con.cursor()
336+
main_cur.execute("BEGIN;")
337+
main_lid = sys.argv[2]
338+
main_data = data_factory.make_trle_tombll_data()
339+
main_soup = scrape_trle.scrape_common.get_soup(
340+
f"https://www.trle.net/sc/levelfeatures.php?lid={main_lid}")
341+
scrape_trle.get_trle_level_card(main_soup, main_data)
342+
add_tombll_json_to_database(main_data, main_con)
343+
main_con.commit()
344+
print(f"File {sys.argv[2]} added successfully.")
345+
333346
elif (sys.argv[1] == "-rm" and number_of_argument == 3):
334347
main_level_id = tombll_read.database_level_id(sys.argv[2], main_con)
335348
main_cur = main_con.cursor()

0 commit comments

Comments
 (0)