Files
USB4VC/user_program/usb4vc_main.py
2022-04-04 00:41:45 +01:00

78 lines
2.4 KiB
Python

import os
import sys
import time
import json
import RPi.GPIO as GPIO
import usb4vc_usb_scan
import usb4vc_ui
import subprocess
# usb4vc_ui.reset_pboard()
def get_current_rpi_model():
current_model = 'Unknown'
try:
current_model = subprocess.getoutput("cat /proc/device-tree/model").replace('\n', '').replace('\r', '').strip()
except Exception as e:
print('get_current_rpi_model:', e)
return current_model
rpi_model_save_file = '/home/pi/rpi_model.txt'
def get_stored_rpi_model():
stored_model = 'Unknown'
try:
with open(rpi_model_save_file, 'r') as file:
stored_model = file.read().replace('\n', '').replace('\r', '').strip()
except Exception as e:
print('get_stored_rpi_model:', e)
return stored_model
def save_rpi_model(model_str):
try:
with open(rpi_model_save_file, 'w') as file:
file.write(model_str)
except Exception as e:
print('save_rpi_model:', e)
def check_rpi_model():
current_model = get_current_rpi_model()
stored_model = get_stored_rpi_model()
print('current_model', current_model)
print('stored_model', stored_model)
if current_model != stored_model:
usb4vc_ui.oled_print_model_changed()
for x in range(20):
print("!!!!!!!!!! DO NOT UNPLUG UNTIL I REBOOT !!!!!!!!!!")
print("!!!!!!!!!! DO NOT UNPLUG UNTIL I REBOOT !!!!!!!!!!")
print("!!!!!!!!!! DO NOT UNPLUG UNTIL I REBOOT !!!!!!!!!!")
time.sleep(0.1)
os.system("cd /home/pi/xpadneo/; sudo ./uninstall.sh")
usb4vc_ui.oled_print_oneline("Recompiling...")
os.system("cd /home/pi/xpadneo/; sudo ./install.sh")
usb4vc_ui.oled_print_reboot()
save_rpi_model(current_model)
time.sleep(2)
os.system('sudo reboot')
time.sleep(10)
check_rpi_model()
usb4vc_ui.ui_init()
usb4vc_ui.ui_thread.start()
usb4vc_usb_scan.usb_device_scan_thread.start()
usb4vc_usb_scan.raw_input_event_parser_thread.start()
while 1:
time.sleep(2)
try:
ertm_status = subprocess.getoutput("cat /sys/module/bluetooth/parameters/disable_ertm").replace('\n', '').replace('\r', '').strip()
if ertm_status != 'Y':
print('ertm_status:', ertm_status)
print("Disabling ERTM....")
subprocess.call('echo 1 > /sys/module/bluetooth/parameters/disable_ertm')
print("DONE")
except Exception:
continue