131 lines
3.8 KiB
Python
131 lines
3.8 KiB
Python
#!/usr/bin/python3
|
|
import requests
|
|
import sys
|
|
import os
|
|
import socket
|
|
import esptool
|
|
|
|
ttgohost = "rdzsonde.local"
|
|
|
|
# usually, rdzsonde.mooo.com should be an alias for that:
|
|
# or, more specifically:
|
|
updatehost = "https://github.com/dl9rdz/rdz_ttgo_sonde/blob/gh-pages/{}/{}?raw=true"
|
|
|
|
screens = ("screens1.txt", "screens2.txt", "screens3.txt")
|
|
allfiles = ("config.txt", "qrg.txt", "networks.txt") + screens
|
|
|
|
optprint = False
|
|
optdir = ""
|
|
|
|
def getfile(name):
|
|
urlg = url+"file/"+name;
|
|
print("Downloading: ",urlg);
|
|
data = requests.get(urlg);
|
|
if optprint:
|
|
print(data.text)
|
|
elif len(data.content)>0:
|
|
f = open(optdir+name, "wb");
|
|
f.write(data.content);
|
|
f.close();
|
|
else:
|
|
print("Error: empty response")
|
|
|
|
def putfile(name):
|
|
print("Uploading: ",optdir+name)
|
|
files = { 'data': (name, open(optdir+name, "rb")), }
|
|
response = requests.post(url+"file", files=files)
|
|
|
|
while len(sys.argv)>=2:
|
|
if sys.argv[1]=="--print":
|
|
del(sys.argv[1])
|
|
optprint = True
|
|
print("Printing file content on screen\n")
|
|
elif sys.argv[1].startswith("--dir="):
|
|
optdir = sys.argv[1][6:]+"/"
|
|
print("Using file directory ",optdir)
|
|
os.makedirs(optdir, exist_ok=True)
|
|
del(sys.argv[1])
|
|
elif sys.argv[1].startswith("--ttgo="):
|
|
ttgohost = sys.argv[1][7:]
|
|
del(sys.argv[1])
|
|
else:
|
|
break
|
|
|
|
if len(sys.argv)<=2:
|
|
print("Usage: ",sys.argv[0]," [--ttgo={ip}] [--print|--dir={dir}] <get|put> <all|config|qrg|networks|screens>");
|
|
print("or: ",sys.argv[0]," <get|put> file {filename}");
|
|
print("or: ",sys.argv[0]," update <devel-xxx|master-yyy>");
|
|
print("or: ",sys.argv[0]," <backup|restore> file.bin");
|
|
print("\n",
|
|
" screens is screens1.txt, screens2.txt, screens3.txt");
|
|
print(" networks is networks.txt (Wifi ssid and password)")
|
|
print(" qrg is qrg.txt (List with scan frequencies)")
|
|
print(" all is screens + network + qrg")
|
|
sys.exit(1)
|
|
|
|
if sys.argv[1]=="backup":
|
|
# backup installed firmware (+ all data) to backup.bin
|
|
sys._argv = sys.argv[:]
|
|
sys.argv=[sys._argv[0],"--chip", "esp32", "--baud", "921600", "--before", "default_reset", "--after", "hard_reset", "read_flash", "0x1000", "0x3FF000", sys.argv[2]]
|
|
esptool.main()
|
|
exit(0)
|
|
|
|
if sys.argv[1]=="restore":
|
|
# restore system from backup.bin
|
|
sys._argv = sys.argv[:]
|
|
sys.argv=[sys._argv[0],"--chip", "esp32", "--baud", "921600", "--before", "default_reset", "--after", "hard_reset", "write_flash", "-z", "--flash_mode", "dio", "--flash_freq", "80m", "--flash_size", "detect", "0x1000", sys.argv[2]]
|
|
esptool.main()
|
|
exit(0)
|
|
|
|
|
|
if sys.argv[1]=="update":
|
|
# update to a new version...
|
|
what = sys.argv[2]
|
|
imgdir = "devel"
|
|
if what.startswith("master"):
|
|
imgdir = "master"
|
|
host = updatehost.format(imgdir, what)
|
|
data = requests.get(host)
|
|
f = open("firmware.bin", "wb")
|
|
f.write(data.content)
|
|
f.close()
|
|
sys._argv = sys.argv[:]
|
|
sys.argv=[sys._argv[0],"--chip", "esp32", "--baud", "921600", "--before", "default_reset", "--after", "hard_reset", "write_flash", "-z", "--flash_mode", "dio", "--flash_freq", "80m", "--flash_size", "detect", "0x1000", "firmware.bin"]
|
|
esptool.main()
|
|
exit(0)
|
|
|
|
addrinfo = socket.gethostbyname(ttgohost)
|
|
url = "http://"+addrinfo+"/"
|
|
print("Using URL ",url)
|
|
|
|
files=()
|
|
|
|
if sys.argv[2]=="file":
|
|
if len(sys.argv)<=3:
|
|
print("get/put file: missing filename\n");
|
|
sys.exit(1);
|
|
files=(sys.argv[3],)
|
|
elif sys.argv[2]=="config":
|
|
files=("config.txt",)
|
|
elif sys.argv[2]=="qrg":
|
|
files=("qrg.txt",)
|
|
elif sys.argv[2]=="networks":
|
|
files=("networks.txt",)
|
|
elif sys.argv[2]=="screens":
|
|
files=screens
|
|
elif sys.argv[2]=="all":
|
|
files=allfiles
|
|
else:
|
|
print("Invalid file specification: ",sys.argv[2])
|
|
sys.exit(1)
|
|
|
|
if(sys.argv[1]=="get"):
|
|
for f in files:
|
|
getfile(f)
|
|
elif(sys.argv[1]=="put"):
|
|
for f in files:
|
|
putfile(f)
|
|
else:
|
|
print("Invalid command ",sys.argv[1])
|
|
|