103 lines
3.0 KiB
Python
103 lines
3.0 KiB
Python
|
# PFEIFER - MAPD - Modified by FrogAi for FrogPilot to automatically update
|
||
|
|
import os
|
||
|
|
import socket
|
||
|
|
import stat
|
||
|
|
import subprocess
|
||
|
|
import time
|
||
|
|
import urllib.error
|
||
|
|
import urllib.request
|
||
|
|
|
||
|
|
from openpilot.common.realtime import Ratekeeper
|
||
|
|
|
||
|
|
VERSION = 'v1'
|
||
|
|
GITHUB_VERSION_URL = f"https://github.com/FrogAi/FrogPilot-Resources/raw/Versions/mapd_version_{VERSION}.txt"
|
||
|
|
GITLAB_VERSION_URL = f"https://gitlab.com/FrogAi/FrogPilot-Resources/-/raw/Versions/mapd_version_{VERSION}.txt"
|
||
|
|
|
||
|
|
MAPD_PATH = '/data/media/0/osm/mapd'
|
||
|
|
VERSION_PATH = '/data/media/0/osm/mapd_version'
|
||
|
|
|
||
|
|
def ping_url(url, timeout=5):
|
||
|
|
no_internet = 0
|
||
|
|
while True:
|
||
|
|
if no_internet > 5:
|
||
|
|
break
|
||
|
|
try:
|
||
|
|
urllib.request.urlopen(url, timeout=timeout)
|
||
|
|
return True
|
||
|
|
except (urllib.error.URLError, socket.timeout):
|
||
|
|
no_internet += 1
|
||
|
|
time.sleep(10)
|
||
|
|
return False
|
||
|
|
|
||
|
|
def get_latest_version():
|
||
|
|
urls = [GITHUB_VERSION_URL, GITLAB_VERSION_URL]
|
||
|
|
for url in urls:
|
||
|
|
try:
|
||
|
|
with urllib.request.urlopen(url) as response:
|
||
|
|
return response.read().decode('utf-8').strip()
|
||
|
|
except Exception as e:
|
||
|
|
print(f"Failed to get latest version from {url}. Error: {e}")
|
||
|
|
print("Failed to get latest version from both sources.")
|
||
|
|
return None
|
||
|
|
|
||
|
|
def download(current_version):
|
||
|
|
urls = [
|
||
|
|
f"https://github.com/pfeiferj/openpilot-mapd/releases/download/{current_version}/mapd",
|
||
|
|
f"https://gitlab.com/FrogAi/openpilot-mapd/-/releases/download/{current_version}/mapd"
|
||
|
|
]
|
||
|
|
|
||
|
|
mapd_dir = os.path.dirname(MAPD_PATH)
|
||
|
|
if not os.path.exists(mapd_dir):
|
||
|
|
os.makedirs(mapd_dir)
|
||
|
|
|
||
|
|
for url in urls:
|
||
|
|
try:
|
||
|
|
with urllib.request.urlopen(url) as f:
|
||
|
|
with open(MAPD_PATH, 'wb') as output:
|
||
|
|
output.write(f.read())
|
||
|
|
os.fsync(output)
|
||
|
|
current_permissions = stat.S_IMODE(os.lstat(MAPD_PATH).st_mode)
|
||
|
|
os.chmod(MAPD_PATH, current_permissions | stat.S_IEXEC)
|
||
|
|
with open(VERSION_PATH, 'w') as output:
|
||
|
|
output.write(current_version)
|
||
|
|
os.fsync(output)
|
||
|
|
print(f"Successfully downloaded mapd from {url}")
|
||
|
|
return
|
||
|
|
except Exception as e:
|
||
|
|
print(f"Failed to download from {url}. Error: {e}")
|
||
|
|
|
||
|
|
print(f"Failed to download mapd for version {current_version} from both sources.")
|
||
|
|
|
||
|
|
def mapd_thread(sm=None, pm=None):
|
||
|
|
rk = Ratekeeper(0.05, print_delay_threshold=None)
|
||
|
|
|
||
|
|
while True:
|
||
|
|
try:
|
||
|
|
if ping_url("https://github.com"):
|
||
|
|
current_version = get_latest_version()
|
||
|
|
if current_version:
|
||
|
|
if not os.path.exists(MAPD_PATH):
|
||
|
|
download(current_version)
|
||
|
|
continue
|
||
|
|
if not os.path.exists(VERSION_PATH):
|
||
|
|
download(current_version)
|
||
|
|
continue
|
||
|
|
with open(VERSION_PATH) as f:
|
||
|
|
content = f.read()
|
||
|
|
if content != current_version:
|
||
|
|
download(current_version)
|
||
|
|
continue
|
||
|
|
|
||
|
|
process = subprocess.Popen(MAPD_PATH)
|
||
|
|
process.wait()
|
||
|
|
except Exception as e:
|
||
|
|
print(e)
|
||
|
|
|
||
|
|
rk.keep_time()
|
||
|
|
|
||
|
|
def main(sm=None, pm=None):
|
||
|
|
mapd_thread(sm, pm)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|