mirror of
https://github.com/DarkFlippers/unleashed-firmware
synced 2024-11-27 15:00:46 +00:00
80 lines
2.7 KiB
Python
80 lines
2.7 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
from os import path
|
||
|
|
|
||
|
|
from flipper.app import App
|
||
|
|
from flipper.utils.fff import *
|
||
|
|
|
||
|
|
|
||
|
|
class Main(App):
|
||
|
|
def init(self):
|
||
|
|
# Subparsers
|
||
|
|
self.subparsers = self.parser.add_subparsers(help="sub-command help")
|
||
|
|
|
||
|
|
self.parser_cleanup = self.subparsers.add_parser(
|
||
|
|
"cleanup", help="Cleanup duplicate remotes"
|
||
|
|
)
|
||
|
|
self.parser_cleanup.add_argument("filename", type=str)
|
||
|
|
self.parser_cleanup.set_defaults(func=self.cleanup)
|
||
|
|
|
||
|
|
def cleanup(self):
|
||
|
|
f = FlipperFormatFile()
|
||
|
|
f.load(self.args.filename)
|
||
|
|
|
||
|
|
filetype, version = f.getHeader()
|
||
|
|
if filetype != "IR library file" or version != 1:
|
||
|
|
self.logger.error(f"Incorrect file type({filetype}) or version({version})")
|
||
|
|
return 1
|
||
|
|
|
||
|
|
data = []
|
||
|
|
unique = {}
|
||
|
|
while True:
|
||
|
|
try:
|
||
|
|
d = {}
|
||
|
|
d["name"] = f.readKey("name")
|
||
|
|
d["type"] = f.readKey("type")
|
||
|
|
key = None
|
||
|
|
if d["type"] == "parsed":
|
||
|
|
d["protocol"] = f.readKey("protocol")
|
||
|
|
d["address"] = f.readKey("address")
|
||
|
|
d["command"] = f.readKey("command")
|
||
|
|
key = f'{d["protocol"]}{d["address"]}{d["command"]}'
|
||
|
|
elif d["type"] == "raw":
|
||
|
|
d["frequency"] = f.readKey("frequency")
|
||
|
|
d["duty_cycle"] = f.readKey("duty_cycle")
|
||
|
|
d["data"] = f.readKey("data")
|
||
|
|
key = f'{d["frequency"]}{d["duty_cycle"]}{d["data"]}'
|
||
|
|
else:
|
||
|
|
raise Exception(f'Unknown type: {d["type"]}')
|
||
|
|
if not key in unique:
|
||
|
|
unique[key] = d
|
||
|
|
data.append(d)
|
||
|
|
else:
|
||
|
|
self.logger.warn(f"Duplicate key: {key}")
|
||
|
|
except EOFError:
|
||
|
|
break
|
||
|
|
# Form new file
|
||
|
|
f = FlipperFormatFile()
|
||
|
|
f.setHeader(filetype, version)
|
||
|
|
for i in data:
|
||
|
|
f.writeComment(None)
|
||
|
|
f.writeKey("name", i["name"])
|
||
|
|
f.writeKey("type", i["type"])
|
||
|
|
if i["type"] == "parsed":
|
||
|
|
f.writeKey("protocol", i["protocol"])
|
||
|
|
f.writeKey("address", i["address"])
|
||
|
|
f.writeKey("command", i["command"])
|
||
|
|
elif i["type"] == "raw":
|
||
|
|
f.writeKey("frequency", i["frequency"])
|
||
|
|
f.writeKey("duty_cycle", i["duty_cycle"])
|
||
|
|
f.writeKey("data", i["data"])
|
||
|
|
else:
|
||
|
|
raise Exception(f'Unknown type: {i["type"]}')
|
||
|
|
f.save(self.args.filename)
|
||
|
|
|
||
|
|
return 0
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
Main()()
|