mirror of
https://github.com/RfidResearchGroup/proxmark3.git
synced 2025-08-14 02:27:26 -07:00
Create update_amiibo_tools_lua.py
This is a python script to automate what the updating of the amiibo_tools.lua file which holds a lua table of all known amiibos. Previously updating the amiibo_tools.lua was a very manual process and prone to errors. This simplifies the process. Signed-off-by: Cory Solovewicz <cory@solovewicz.com>
This commit is contained in:
parent
5b53bf803d
commit
3939bcb541
1 changed files with 114 additions and 0 deletions
114
client/update_amiibo_tools_lua.py
Normal file
114
client/update_amiibo_tools_lua.py
Normal file
|
@ -0,0 +1,114 @@
|
|||
#!/usr/bin/env python3
|
||||
|
||||
"""
|
||||
-----------------------------------------------------------------------------
|
||||
Name: update_amiibo_tools_lua.py
|
||||
|
||||
Author: Cory Solovewicz
|
||||
|
||||
Description:
|
||||
This is a python script to automate what the updating of the amiibo_tools.lua
|
||||
file which holds a lua table of all known amiibos. Previously updating the
|
||||
amiibo_tools.lua was a very manual process.
|
||||
|
||||
This script automates the following original command:
|
||||
curl https://raw.githubusercontent.com/N3evin/AmiiboAPI/master/database/amiibo.json | jq 'del(.amiibos[].release)' | jq 'del(.characters)' | pbcopy --> transform to table
|
||||
And outputs the formatted file as amiibo_tools.lua
|
||||
If everything goes well, this should be an updated copy of amiibo_tools.lua
|
||||
which can then be placed in the /lualibs/ directory.
|
||||
The temporary amiibo.json file is then deleted
|
||||
|
||||
Dependencies:
|
||||
python3 -m pip install jq
|
||||
|
||||
How to run:
|
||||
python update_amiibo_tools_lua.py
|
||||
The script will create the file amiibo_tools.lua
|
||||
|
||||
After running, manually backup the original /lualibs/amiibo_tools.lua and move the
|
||||
updated amiibo_tools.lua to the /lualibs/ directory.
|
||||
-----------------------------------------------------------------------------
|
||||
"""
|
||||
|
||||
import os
|
||||
import re
|
||||
import subprocess
|
||||
import json
|
||||
from jq import jq
|
||||
|
||||
def fetch_data():
|
||||
print("Fetching amiibo.json")
|
||||
# Perform the curl command
|
||||
curl_command = "curl https://raw.githubusercontent.com/N3evin/AmiiboAPI/master/database/amiibo.json"
|
||||
curl_process = subprocess.Popen(curl_command.split(), stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
||||
output, error = curl_process.communicate()
|
||||
|
||||
if curl_process.returncode != 0:
|
||||
print("Error fetching data: ", error.decode())
|
||||
return None
|
||||
return output
|
||||
|
||||
def filter_data(data):
|
||||
print("Filtering downloaded data")
|
||||
# Convert the output to JSON and use jq to filter data
|
||||
data_json = json.loads(data)
|
||||
filtered_data_json = jq('del(.amiibos[].release) | del(.characters)').transform(data_json)
|
||||
# Convert the filtered JSON data back to a string, preserving Unicode characters
|
||||
filtered_data = json.dumps(filtered_data_json, indent=2, ensure_ascii=False)
|
||||
return filtered_data
|
||||
|
||||
def save_data(filtered_data, filename):
|
||||
# Save filtered data to file
|
||||
with open(filename, 'w', encoding='utf-8') as file:
|
||||
file.write(filtered_data)
|
||||
print(f"Data saved to {filename}")
|
||||
|
||||
def process_file(filename):
|
||||
# Open the file
|
||||
with open(filename, 'r', encoding='utf-8') as file:
|
||||
data = file.read()
|
||||
|
||||
# Perform the replacements
|
||||
data = data.replace('"name"', 'name')
|
||||
data = data.replace('"amiibo_series"', 'amiibo_series')
|
||||
data = data.replace('"amiibos"', 'amiibos')
|
||||
data = data.replace('"game_series"', 'game_series')
|
||||
data = data.replace('"types"', 'types')
|
||||
data = data.replace(':', ' =')
|
||||
data = re.sub('"0x', '["0x', data)
|
||||
data = re.sub('" =', '"] =', data)
|
||||
|
||||
# Prepend the text
|
||||
prepend_text = 'local amiibo_tools = {}\n\n-- curl https://raw.githubusercontent.com/N3evin/AmiiboAPI/master/database/amiibo.json | jq \'del(.amiibos[].release)\' | jq \'del(.characters)\' | pbcopy --> transform to table\namiibo_tools.db =\n'
|
||||
data = prepend_text + data
|
||||
|
||||
# Append the text
|
||||
append_text = '\n\nreturn amiibo_tools\n'
|
||||
data = data + append_text
|
||||
|
||||
return data
|
||||
|
||||
def write_to_file(data, filename):
|
||||
# Write the output
|
||||
with open(filename, 'w', encoding='utf-8') as file:
|
||||
file.write(data)
|
||||
print(f"Output written to {filename}")
|
||||
|
||||
def delete_file(filename):
|
||||
try:
|
||||
os.remove(filename)
|
||||
print(f"Temporary file {filename} deleted")
|
||||
except OSError as e:
|
||||
print(f"Error deleting file {filename}: ", e)
|
||||
|
||||
def main():
|
||||
data = fetch_data()
|
||||
if data:
|
||||
filtered_data = filter_data(data)
|
||||
save_data(filtered_data, 'amiibo.json')
|
||||
processed_data = process_file('amiibo.json')
|
||||
write_to_file(processed_data, 'amiibo_tools.lua')
|
||||
delete_file('amiibo.json')
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
Loading…
Add table
Add a link
Reference in a new issue