__init__.py 11 KB

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