mirror of
https://github.com/clockworkpi/launcher.git
synced 2025-12-12 17:58:50 +01:00
350 lines
11 KiB
Python
350 lines
11 KiB
Python
# -*- coding: utf-8 -*-
|
|
import os
|
|
import pygame
|
|
import platform
|
|
import commands
|
|
#import glob
|
|
#from beeprint import pp
|
|
from libs.roundrects import aa_round_rect
|
|
|
|
## local UI import
|
|
from UI.constants import Width,Height,ICON_TYPES,RESTARTUI
|
|
from UI.page import Page,PageSelector
|
|
from UI.label import Label
|
|
from UI.util_funcs import midRect,FileExists
|
|
from UI.keys_def import CurKeys, IsKeyStartOrA, IsKeyMenuOrB
|
|
from UI.scroller import ListScroller
|
|
from UI.icon_pool import MyIconPool
|
|
from UI.icon_item import IconItem
|
|
from UI.multilabel import MultiLabel
|
|
from UI.skin_manager import MySkinManager
|
|
from UI.lang_manager import MyLangManager
|
|
from UI.info_page_list_item import InfoPageListItem
|
|
from UI.info_page_selector import InfoPageSelector
|
|
|
|
from libs.DBUS import is_wifi_connected_now
|
|
import config
|
|
|
|
class ListPageSelector(InfoPageSelector):
|
|
def Draw(self):
|
|
idx = self._Parent._PsIndex
|
|
|
|
if idx < len(self._Parent._MyList):
|
|
x = self._Parent._MyList[idx]._PosX+2 ## ++
|
|
y = self._Parent._MyList[idx]._PosY+1
|
|
h = self._Parent._MyList[idx]._Height -3
|
|
|
|
self._PosX = x
|
|
self._PosY = y
|
|
self._Height = h
|
|
|
|
aa_round_rect(self._Parent._CanvasHWND,
|
|
(x,y,self._Width-4,h),self._BackgroundColor,4,0,self._BackgroundColor)
|
|
|
|
|
|
class PageListItem(InfoPageListItem):
|
|
_PosX = 0
|
|
_PosY = 0
|
|
_Width = 0
|
|
_Height = 30
|
|
|
|
_Labels = {}
|
|
_Icons = {}
|
|
_Fonts = {}
|
|
|
|
_LinkObj = None
|
|
|
|
_Active = False
|
|
_Value = ""
|
|
|
|
def Draw(self):
|
|
|
|
self._Labels["Text"]._PosY = self._PosY+ (self._Height- self._Labels["Text"]._Height)/2
|
|
|
|
if self._Active == True:
|
|
self._Parent._Icons["done"].NewCoord( self._Parent._Width-30,self._PosY+5)
|
|
self._Parent._Icons["done"].Draw()
|
|
|
|
self._Labels["Text"].Draw(self._Active)
|
|
|
|
if "Small" in self._Labels:
|
|
self._Labels["Small"]._PosX = self._Width - self._Labels["Small"]._Width -10
|
|
self._Labels["Small"]._PosY = self._PosY + (self._Height- self._Labels["Small"]._Height)/2
|
|
self._Labels["Small"].Draw()
|
|
|
|
pygame.draw.line(self._Parent._CanvasHWND,MySkinManager.GiveColor('Line'),(self._PosX,self._PosY+self._Height-1),(self._PosX+self._Width,self._PosY+self._Height-1),1)
|
|
|
|
class GateWayPage(Page):
|
|
_FootMsg = ["Nav","Clear All","","Back","Select"]
|
|
_MyList = []
|
|
_ListFont = MyLangManager.TrFont("notosanscjk15")
|
|
|
|
_AList = {}
|
|
|
|
_Scrolled = 0
|
|
|
|
_BGwidth = 320
|
|
_BGheight = 240-24-20
|
|
|
|
_DrawOnce = False
|
|
_Scroller = None
|
|
_InfoPage = None
|
|
|
|
def __init__(self):
|
|
Page.__init__(self)
|
|
self._Icons = {}
|
|
|
|
def GenList(self):
|
|
|
|
self._MyList = []
|
|
|
|
start_x = 0
|
|
start_y = 0
|
|
last_height = 0
|
|
|
|
drivers = [["usb0","USB Ethernet"],
|
|
["wlan0","Wi-Fi"]]
|
|
|
|
|
|
for i,u in enumerate( drivers ):
|
|
#print(i,u)
|
|
li = PageListItem()
|
|
li._Parent = self
|
|
li._PosX = start_x
|
|
li._PosY = start_y + last_height
|
|
li._Width = Width
|
|
li._Fonts["normal"] = self._ListFont
|
|
li._Active = False
|
|
li._Value = u[0]
|
|
li.Init( u[1] )
|
|
|
|
last_height += li._Height
|
|
|
|
self._MyList.append(li)
|
|
|
|
def Init(self):
|
|
if self._Screen != None:
|
|
if self._Screen._CanvasHWND != None and self._CanvasHWND == None:
|
|
self._HWND = self._Screen._CanvasHWND
|
|
self._CanvasHWND = pygame.Surface( (self._Screen._Width,self._BGheight) )
|
|
|
|
self._PosX = self._Index*self._Screen._Width
|
|
self._Width = self._Screen._Width ## equal to screen width
|
|
self._Height = self._Screen._Height
|
|
|
|
done = IconItem()
|
|
done._ImgSurf = MyIconPool._Icons["done"]
|
|
done._MyType = ICON_TYPES["STAT"]
|
|
done._Parent = self
|
|
self._Icons["done"] = done
|
|
|
|
ps = ListPageSelector()
|
|
ps._Parent = self
|
|
self._Ps = ps
|
|
self._PsIndex = 0
|
|
|
|
self.GenList()
|
|
|
|
self._Scroller = ListScroller()
|
|
self._Scroller._Parent = self
|
|
self._Scroller._PosX = self._Width - 10
|
|
self._Scroller._PosY = 2
|
|
self._Scroller.Init()
|
|
self._Scroller.SetCanvasHWND(self._HWND)
|
|
|
|
def Click(self):
|
|
if len(self._MyList) == 0:
|
|
return
|
|
|
|
cur_li = self._MyList[self._PsIndex]
|
|
if cur_li._Active == True:
|
|
out = commands.getstatusoutput("sudo ip route | grep default | cut -d \" \" -f3")
|
|
if len(out[1]) > 7:
|
|
self._Screen._MsgBox.SetText(out[1])
|
|
self._Screen._MsgBox.Draw()
|
|
self._Screen.SwapAndShow()
|
|
return
|
|
|
|
print(cur_li._Value)
|
|
|
|
if "arm" in platform.machine():
|
|
for i in self._MyList:
|
|
i._Active = False
|
|
|
|
self._Screen._MsgBox.SetText("Applying")
|
|
self._Screen._MsgBox.Draw()
|
|
self._Screen.SwapAndShow()
|
|
|
|
cur_li._Active = self.ApplyGateWay(cur_li._Value)
|
|
|
|
pygame.time.delay(1000)
|
|
self._Screen.Draw()
|
|
self._Screen.SwapAndShow()
|
|
else:
|
|
self._Screen._MsgBox.SetText("Do it in GameShell")
|
|
self._Screen._MsgBox.Draw()
|
|
self._Screen.SwapAndShow()
|
|
|
|
def ClearAllGateways(self):
|
|
self._Screen._MsgBox.SetText("Cleaning up")
|
|
self._Screen._MsgBox.Draw()
|
|
self._Screen.SwapAndShow()
|
|
os.system("sudo ip route del 0/0")
|
|
pygame.time.delay(800)
|
|
|
|
for i in self._MyList:
|
|
i._Active = False
|
|
|
|
self._Screen.Draw()
|
|
self._Screen.SwapAndShow()
|
|
|
|
def ApplyGateWay(self,gateway):
|
|
os.system("sudo ip route del 0/0")
|
|
if gateway== "usb0":
|
|
out = commands.getstatusoutput("sudo ifconfig usb0 | grep inet | tr -s \" \"| cut -d \" \" -f3")
|
|
if len(out[1]) > 7:
|
|
if "error" not in out[1]:
|
|
parts = out[1].split(".")
|
|
if len(parts) == 4:##IPv4
|
|
tp3 = int(parts[3])
|
|
tmp = tp3
|
|
if tp3 == 0:
|
|
tmp = int(parts[3]) + 1
|
|
elif tp3 == 1:
|
|
tmp = int(parts[3]) + 1
|
|
elif tp3 > 1:
|
|
tmp = int(parts[3]) - 1
|
|
|
|
parts[3] = str(tmp)
|
|
ipaddress = ".".join(parts)
|
|
os.system("sudo route add default gw "+ipaddress)
|
|
return True
|
|
else:
|
|
if is_wifi_connected_now():
|
|
os.system("sudo dhclient wlan0")
|
|
return True
|
|
else:
|
|
self._Screen._MsgBox.SetText("Wi-Fi is not connected")
|
|
self._Screen._MsgBox.Draw()
|
|
self._Screen.SwapAndShow()
|
|
return False
|
|
|
|
return False
|
|
|
|
|
|
def OnLoadCb(self):
|
|
self._Scrolled = 0
|
|
self._PosY = 0
|
|
self._DrawOnce = False
|
|
|
|
## grep Driver /etc/xorg.conf | tr -s " " | cut -d " " -f3
|
|
## "fbturbo"
|
|
## "modesetting"
|
|
thedrv = ""
|
|
|
|
if "arm" in platform.machine():
|
|
out = commands.getstatusoutput("sudo ip route | grep default")
|
|
if len(out[1]) > 7:
|
|
if "usb0" in out[1]:
|
|
thedrv = "usb0"
|
|
elif "wlan0" in out[1]:
|
|
thedrv = "wlan0"
|
|
|
|
for i in self._MyList:
|
|
i._Active = False
|
|
|
|
if thedrv != "":
|
|
for i in self._MyList:
|
|
if thedrv in i._Value:
|
|
i._Active = True
|
|
## if usb0 and wlan0 all actived, clear all
|
|
|
|
def OnReturnBackCb(self):
|
|
pass
|
|
"""
|
|
self.ReturnToUpLevelPage()
|
|
self._Screen.Draw()
|
|
self._Screen.SwapAndShow()
|
|
"""
|
|
def KeyDown(self,event):
|
|
if IsKeyMenuOrB(event.key):
|
|
self.ReturnToUpLevelPage()
|
|
self._Screen.Draw()
|
|
self._Screen.SwapAndShow()
|
|
|
|
if IsKeyStartOrA(event.key):
|
|
self.Click()
|
|
|
|
if event.key == CurKeys["Y"]:
|
|
self.ClearAllGateways()
|
|
|
|
if event.key == CurKeys["Up"]:
|
|
self.ScrollUp()
|
|
self._Screen.Draw()
|
|
self._Screen.SwapAndShow()
|
|
if event.key == CurKeys["Down"]:
|
|
self.ScrollDown()
|
|
self._Screen.Draw()
|
|
self._Screen.SwapAndShow()
|
|
|
|
|
|
def Draw(self):
|
|
|
|
self.ClearCanvas()
|
|
if len(self._MyList) == 0:
|
|
return
|
|
|
|
else:
|
|
if len(self._MyList) * PageListItem._Height > self._Height:
|
|
self._Ps._Width = self._Width - 11
|
|
self._Ps.Draw()
|
|
for i in self._MyList:
|
|
if i._PosY > self._Height + self._Height/2:
|
|
break
|
|
if i._PosY < 0:
|
|
continue
|
|
i.Draw()
|
|
self._Scroller.UpdateSize( len(self._MyList)*PageListItem._Height, self._PsIndex*PageListItem._Height)
|
|
self._Scroller.Draw()
|
|
|
|
else:
|
|
self._Ps._Width = self._Width
|
|
self._Ps.Draw()
|
|
for i in self._MyList:
|
|
if i._PosY > self._Height + self._Height/2:
|
|
break
|
|
if i._PosY < 0:
|
|
continue
|
|
i.Draw()
|
|
|
|
if self._HWND != None:
|
|
self._HWND.fill((255,255,255))
|
|
|
|
self._HWND.blit(self._CanvasHWND,(self._PosX,self._PosY,self._Width, self._Height ) )
|
|
|
|
|
|
class APIOBJ(object):
|
|
|
|
_Page = None
|
|
def __init__(self):
|
|
pass
|
|
def Init(self,main_screen):
|
|
self._Page = GateWayPage()
|
|
self._Page._Screen = main_screen
|
|
self._Page._Name ="Gateway switch"
|
|
self._Page.Init()
|
|
|
|
def API(self,main_screen):
|
|
if main_screen !=None:
|
|
main_screen.PushPage(self._Page)
|
|
main_screen.Draw()
|
|
main_screen.SwapAndShow()
|
|
|
|
OBJ = APIOBJ()
|
|
def Init(main_screen):
|
|
OBJ.Init(main_screen)
|
|
def API(main_screen):
|
|
OBJ.API(main_screen)
|
|
|
|
|