__init__.py 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350
  1. # -*- coding: utf-8 -*-
  2. import os
  3. import pygame
  4. import platform
  5. import commands
  6. #import glob
  7. #from beeprint import pp
  8. from libs.roundrects import aa_round_rect
  9. ## local UI import
  10. from UI.constants import Width,Height,ICON_TYPES,RESTARTUI
  11. from UI.page import Page,PageSelector
  12. from UI.label import Label
  13. from UI.fonts import fonts
  14. from UI.util_funcs import midRect,FileExists
  15. from UI.keys_def import CurKeys
  16. from UI.scroller import ListScroller
  17. from UI.icon_pool import MyIconPool
  18. from UI.icon_item import IconItem
  19. from UI.multilabel import MultiLabel
  20. from UI.skin_manager import MySkinManager
  21. from UI.lang_manager import MyLangManager
  22. from UI.info_page_list_item import InfoPageListItem
  23. from UI.info_page_selector import InfoPageSelector
  24. from libs.DBUS import is_wifi_connected_now
  25. import config
  26. class ListPageSelector(InfoPageSelector):
  27. def Draw(self):
  28. idx = self._Parent._PsIndex
  29. if idx < len(self._Parent._MyList):
  30. x = self._Parent._MyList[idx]._PosX+2 ## ++
  31. y = self._Parent._MyList[idx]._PosY+1
  32. h = self._Parent._MyList[idx]._Height -3
  33. self._PosX = x
  34. self._PosY = y
  35. self._Height = h
  36. aa_round_rect(self._Parent._CanvasHWND,
  37. (x,y,self._Width-4,h),self._BackgroundColor,4,0,self._BackgroundColor)
  38. class PageListItem(InfoPageListItem):
  39. _PosX = 0
  40. _PosY = 0
  41. _Width = 0
  42. _Height = 30
  43. _Labels = {}
  44. _Icons = {}
  45. _Fonts = {}
  46. _LinkObj = None
  47. _Active = False
  48. _Value = ""
  49. def Draw(self):
  50. self._Labels["Text"]._PosY = self._PosY+ (self._Height- self._Labels["Text"]._Height)/2
  51. if self._Active == True:
  52. self._Parent._Icons["done"].NewCoord( self._Parent._Width-30,self._PosY+5)
  53. self._Parent._Icons["done"].Draw()
  54. self._Labels["Text"].Draw(self._Active)
  55. if "Small" in self._Labels:
  56. self._Labels["Small"]._PosX = self._Width - self._Labels["Small"]._Width -10
  57. self._Labels["Small"]._PosY = self._PosY + (self._Height- self._Labels["Small"]._Height)/2
  58. self._Labels["Small"].Draw()
  59. 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)
  60. class GateWayPage(Page):
  61. _FootMsg = ["Nav","","Clear All","Back","Select"]
  62. _MyList = []
  63. _ListFont = fonts["notosanscjk15"]
  64. _AList = {}
  65. _Scrolled = 0
  66. _BGwidth = 320
  67. _BGheight = 240-24-20
  68. _DrawOnce = False
  69. _Scroller = None
  70. _InfoPage = None
  71. def __init__(self):
  72. Page.__init__(self)
  73. self._Icons = {}
  74. def GenList(self):
  75. self._MyList = []
  76. start_x = 0
  77. start_y = 0
  78. last_height = 0
  79. drivers = [["usb0","USB Ethernet"],
  80. ["wlan0","Wi-Fi"]]
  81. for i,u in enumerate( drivers ):
  82. #print(i,u)
  83. li = PageListItem()
  84. li._Parent = self
  85. li._PosX = start_x
  86. li._PosY = start_y + last_height
  87. li._Width = Width
  88. li._Fonts["normal"] = self._ListFont
  89. li._Active = False
  90. li._Value = u[0]
  91. li.Init( u[1] )
  92. last_height += li._Height
  93. self._MyList.append(li)
  94. def Init(self):
  95. if self._Screen != None:
  96. if self._Screen._CanvasHWND != None and self._CanvasHWND == None:
  97. self._HWND = self._Screen._CanvasHWND
  98. self._CanvasHWND = pygame.Surface( (self._Screen._Width,self._BGheight) )
  99. self._PosX = self._Index*self._Screen._Width
  100. self._Width = self._Screen._Width ## equal to screen width
  101. self._Height = self._Screen._Height
  102. done = IconItem()
  103. done._ImgSurf = MyIconPool._Icons["done"]
  104. done._MyType = ICON_TYPES["STAT"]
  105. done._Parent = self
  106. self._Icons["done"] = done
  107. ps = ListPageSelector()
  108. ps._Parent = self
  109. self._Ps = ps
  110. self._PsIndex = 0
  111. self.GenList()
  112. self._Scroller = ListScroller()
  113. self._Scroller._Parent = self
  114. self._Scroller._PosX = self._Width - 10
  115. self._Scroller._PosY = 2
  116. self._Scroller.Init()
  117. self._Scroller.SetCanvasHWND(self._HWND)
  118. def Click(self):
  119. if len(self._MyList) == 0:
  120. return
  121. cur_li = self._MyList[self._PsIndex]
  122. if cur_li._Active == True:
  123. out = commands.getstatusoutput("sudo ip route | grep default | cut -d \" \" -f3")
  124. if len(out[1]) > 7:
  125. self._Screen._MsgBox.SetText(out[1])
  126. self._Screen._MsgBox.Draw()
  127. self._Screen.SwapAndShow()
  128. return
  129. print(cur_li._Value)
  130. if "arm" in platform.machine():
  131. for i in self._MyList:
  132. i._Active = False
  133. self._Screen._MsgBox.SetText("Applying")
  134. self._Screen._MsgBox.Draw()
  135. self._Screen.SwapAndShow()
  136. cur_li._Active = self.ApplyGateWay(cur_li._Value)
  137. pygame.time.delay(1000)
  138. self._Screen.Draw()
  139. self._Screen.SwapAndShow()
  140. else:
  141. self._Screen._MsgBox.SetText("Do it in GameShell")
  142. self._Screen._MsgBox.Draw()
  143. self._Screen.SwapAndShow()
  144. def ClearAllGateways(self):
  145. self._Screen._MsgBox.SetText("Cleaning up")
  146. self._Screen._MsgBox.Draw()
  147. self._Screen.SwapAndShow()
  148. os.system("sudo ip route del 0/0")
  149. pygame.time.delay(800)
  150. for i in self._MyList:
  151. i._Active = False
  152. self._Screen.Draw()
  153. self._Screen.SwapAndShow()
  154. def ApplyGateWay(self,gateway):
  155. os.system("sudo ip route del 0/0")
  156. if gateway== "usb0":
  157. out = commands.getstatusoutput("sudo ifconfig usb0 | grep inet | tr -s \" \"| cut -d \" \" -f3")
  158. if len(out[1]) > 7:
  159. if "error" not in out[1]:
  160. parts = out[1].split(".")
  161. if len(parts) == 4:##IPv4
  162. tp3 = int(parts[3])
  163. tmp = tp3
  164. if tp3 == 0:
  165. tmp = int(parts[3]) + 1
  166. elif tp3 == 1:
  167. tmp = int(parts[3]) + 1
  168. elif tp3 > 1:
  169. tmp = int(parts[3]) - 1
  170. parts[3] = str(tmp)
  171. ipaddress = ".".join(parts)
  172. os.system("sudo route add default gw "+ipaddress)
  173. return True
  174. else:
  175. if is_wifi_connected_now():
  176. os.system("sudo dhclient wlan0")
  177. return True
  178. else:
  179. self._Screen._MsgBox.SetText("Wi-Fi is not connected")
  180. self._Screen._MsgBox.Draw()
  181. self._Screen.SwapAndShow()
  182. return False
  183. return False
  184. def OnLoadCb(self):
  185. self._Scrolled = 0
  186. self._PosY = 0
  187. self._DrawOnce = False
  188. ## grep Driver /etc/xorg.conf | tr -s " " | cut -d " " -f3
  189. ## "fbturbo"
  190. ## "modesetting"
  191. thedrv = ""
  192. if "arm" in platform.machine():
  193. out = commands.getstatusoutput("sudo ip route | grep default")
  194. if len(out[1]) > 7:
  195. if "usb0" in out[1]:
  196. thedrv = "usb0"
  197. elif "wlan0" in out[1]:
  198. thedrv = "wlan0"
  199. for i in self._MyList:
  200. i._Active = False
  201. if thedrv != "":
  202. for i in self._MyList:
  203. if thedrv in i._Value:
  204. i._Active = True
  205. ## if usb0 and wlan0 all actived, clear all
  206. def OnReturnBackCb(self):
  207. pass
  208. """
  209. self.ReturnToUpLevelPage()
  210. self._Screen.Draw()
  211. self._Screen.SwapAndShow()
  212. """
  213. def KeyDown(self,event):
  214. if event.key == CurKeys["A"] or event.key == CurKeys["Menu"]:
  215. self.ReturnToUpLevelPage()
  216. self._Screen.Draw()
  217. self._Screen.SwapAndShow()
  218. if event.key == CurKeys["B"]:
  219. self.Click()
  220. if event.key == CurKeys["Y"]:
  221. self.ClearAllGateways()
  222. if event.key == CurKeys["Up"]:
  223. self.ScrollUp()
  224. self._Screen.Draw()
  225. self._Screen.SwapAndShow()
  226. if event.key == CurKeys["Down"]:
  227. self.ScrollDown()
  228. self._Screen.Draw()
  229. self._Screen.SwapAndShow()
  230. def Draw(self):
  231. self.ClearCanvas()
  232. if len(self._MyList) == 0:
  233. return
  234. else:
  235. if len(self._MyList) * PageListItem._Height > self._Height:
  236. self._Ps._Width = self._Width - 11
  237. self._Ps.Draw()
  238. for i in self._MyList:
  239. if i._PosY > self._Height + self._Height/2:
  240. break
  241. if i._PosY < 0:
  242. continue
  243. i.Draw()
  244. self._Scroller.UpdateSize( len(self._MyList)*PageListItem._Height, self._PsIndex*PageListItem._Height)
  245. self._Scroller.Draw()
  246. else:
  247. self._Ps._Width = self._Width
  248. self._Ps.Draw()
  249. for i in self._MyList:
  250. if i._PosY > self._Height + self._Height/2:
  251. break
  252. if i._PosY < 0:
  253. continue
  254. i.Draw()
  255. if self._HWND != None:
  256. self._HWND.fill((255,255,255))
  257. self._HWND.blit(self._CanvasHWND,(self._PosX,self._PosY,self._Width, self._Height ) )
  258. class APIOBJ(object):
  259. _Page = None
  260. def __init__(self):
  261. pass
  262. def Init(self,main_screen):
  263. self._Page = GateWayPage()
  264. self._Page._Screen = main_screen
  265. self._Page._Name ="Gateway switch"
  266. self._Page.Init()
  267. def API(self,main_screen):
  268. if main_screen !=None:
  269. main_screen.PushPage(self._Page)
  270. main_screen.Draw()
  271. main_screen.SwapAndShow()
  272. OBJ = APIOBJ()
  273. def Init(main_screen):
  274. OBJ.Init(main_screen)
  275. def API(main_screen):
  276. OBJ.API(main_screen)