__init__.py 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731
  1. # -*- coding: utf-8 -*-
  2. import re
  3. import pygame
  4. #import math
  5. import commands
  6. import dbus
  7. from beeprint import pp
  8. from libs.roundrects import aa_round_rect
  9. from libs.DBUS import bus, adapter,devices
  10. #import gobject
  11. #from wicd import misc
  12. ## local UI import
  13. from UI.constants import Width,Height,ICON_TYPES
  14. from UI.page import Page,PageSelector
  15. from UI.label import Label
  16. from UI.fonts import fonts
  17. from UI.util_funcs import midRect
  18. from UI.keys_def import CurKeys
  19. from UI.scroller import ListScroller
  20. from UI.icon_pool import MyIconPool
  21. from UI.icon_item import IconItem
  22. from UI.multi_icon_item import MultiIconItem
  23. from UI.skin_manager import MySkinManager
  24. from UI.confirm_page import ConfirmPage
  25. from UI.info_page_list_item import InfoPageListItem
  26. from UI.multilabel import MultiLabel
  27. from UI.lang_manager import MyLangManager
  28. from net_item import NetItem
  29. class BleForgetConfirmPage(ConfirmPage):
  30. _ConfirmText = MyLangManager.Tr("ConfirmForgetQ")
  31. def KeyDown(self,event):
  32. if event.key == CurKeys["Menu"] or event.key == CurKeys["A"]:
  33. self.ReturnToUpLevelPage()
  34. self._Screen.Draw()
  35. self._Screen.SwapAndShow()
  36. if event.key == CurKeys["B"]:
  37. self.SnapMsg(MyLangManager.Tr("Deleting"))
  38. self._Screen.Draw()
  39. self._Screen.SwapAndShow()
  40. try:
  41. #self._Parent._Adapter.RemoveDevice()
  42. print("try to RemoveDevice")
  43. except Exception,e:
  44. print(str(e))
  45. pygame.time.delay(400)
  46. self.ReturnToUpLevelPage()
  47. self._Screen.Draw()
  48. self._Screen.SwapAndShow()
  49. def Draw(self):
  50. #self.ClearCanvas()
  51. self.DrawBG()
  52. for i in self._MyList:
  53. i.Draw()
  54. self.Reset()
  55. class BleInfoPageSelector(PageSelector):
  56. _BackgroundColor = MySkinManager.GiveColor('Front')
  57. def __init__(self):
  58. self._PosX = 0
  59. self._PosY = 0
  60. self._Height = 0
  61. def AnimateDraw(self,x2,y2):
  62. pass
  63. def Draw(self):
  64. idx = self._Parent._PsIndex
  65. if idx < len( self._Parent._MyList):
  66. x = self._PosX+2
  67. y = self._Parent._MyList[idx]._PosY+1
  68. h = self._Parent._MyList[idx]._Height -3
  69. self._PosX = x
  70. self._PosY = y
  71. self._Height = h
  72. aa_round_rect(self._Parent._CanvasHWND,
  73. (x,y,self._Width-4,h),self._BackgroundColor,4,0,self._BackgroundColor)
  74. class BleInfoPage(Page):
  75. _FootMsg = ["Nav","Disconnect","Forget","Back",""]
  76. _MyList = []
  77. _ListFontObj = MyLangManager.TrFont("varela15")
  78. _ListSmFontObj = fonts["varela12"] # small font
  79. _ListSm2FontObj= fonts["varela11"]
  80. _AList = {}
  81. _Path = ""
  82. def Init(self):
  83. if self._Screen != None:
  84. if self._Screen._CanvasHWND != None and self._CanvasHWND == None:
  85. self._CanvasHWND = self._Screen._CanvasHWND
  86. self._PosX = self._Index*self._Screen._Width
  87. self._Width = self._Screen._Width ## equal to screen width
  88. self._Height = self._Screen._Height
  89. ps = BleInfoPageSelector()
  90. ps._Parent = self
  91. self._Ps = ps
  92. self._PsIndex = 0
  93. #_AList is an object
  94. self.GenList()
  95. self._Scroller = ListScroller()
  96. self._Scroller._Parent = self
  97. self._Scroller._PosX = 2
  98. self._Scroller._PosY = 2
  99. self._Scroller.Init()
  100. self._ConfirmPage1 = BleForgetConfirmPage()
  101. self._ConfirmPage1._Screen = self._Screen
  102. self._ConfirmPage1._Name = "ConfirmForget"
  103. self._ConfirmPage1._Parent = self
  104. self._ConfirmPage1.Init()
  105. def GenList(self):
  106. if self._AList== None:
  107. return
  108. self._MyList = []
  109. self._PsIndex = 0
  110. start_x = 0
  111. start_y = 0
  112. for i,v in enumerate( self._AList):
  113. #print(i,v) # (0, dbus.String(u'AddressType'))
  114. li = InfoPageListItem()
  115. li._Parent = self
  116. li._PosX = start_x
  117. li._PosY = start_y + i*InfoPageListItem._Height
  118. li._Width = Width
  119. li._Fonts["normal"] = self._ListFontObj
  120. if v == "UUIDs":
  121. li._Fonts["small"] = self._ListSm2FontObj
  122. else:
  123. li._Fonts["small"] = self._ListSmFontObj
  124. li.Init( str(v) )
  125. li._Flag = v
  126. if v =="UUIDs":
  127. if len(self._AList[v]) > 1:
  128. pp(self._AList[v][0])
  129. sm_text = str(self._AList[v][0])
  130. else:
  131. sm_text = "<empty>"
  132. else:
  133. sm_text = str(self._AList[v])
  134. if sm_text == "0":
  135. sm_text="No"
  136. elif sm_text == "1":
  137. sm_text="Yes"
  138. sm_text = sm_text[:20]
  139. li.SetSmallText(sm_text)
  140. li._PosX = 2
  141. self._MyList.append(li)
  142. def TryToForget(self):
  143. global adapter
  144. proxy_obj = bus.get_object("org.bluez", self._Path)
  145. dev = dbus.Interface(proxy_obj, "org.bluez.Device1")
  146. self._Screen._MsgBox.SetText("Forgeting")
  147. self._Screen._MsgBox.Draw()
  148. self._Screen.SwapAndShow()
  149. try:
  150. adapter.RemoveDevice(dev)
  151. except Exception,e:
  152. print(str(e))
  153. pygame.time.delay(400)
  154. self.ReturnToUpLevelPage()
  155. self._Screen.Draw()
  156. self._Screen.SwapAndShow()
  157. def TryToDisconnect(self):
  158. global bus
  159. if "Connected" in self._AList:
  160. if self._AList["Connected"] == 0:
  161. return
  162. proxy_obj = bus.get_object("org.bluez", self._Path)
  163. dev = dbus.Interface(proxy_obj, "org.bluez.Device1")
  164. self._Screen._FootBar.UpdateNavText("Disconnecting")
  165. self._Screen._MsgBox.SetText("Disconnecting")
  166. self._Screen._MsgBox.Draw()
  167. self._Screen.SwapAndShow()
  168. try:
  169. dev.Disconnect()
  170. except Exception,e:
  171. print(str(e))
  172. pygame.time.delay(300)
  173. self.ReturnToUpLevelPage()
  174. self._Screen.Draw()
  175. self._Screen.SwapAndShow()
  176. self._Screen._FootBar.ResetNavText()
  177. def Click(self):
  178. if self._PsIndex >= len(self._MyList):
  179. return
  180. cur_li = self._MyList[self._PsIndex]
  181. print(cur_li._Flag)
  182. if cur_li._Flag in self._AList:
  183. print(self._AList[ cur_li._Flag ])
  184. def OnLoadCb(self):
  185. if self._AList != None:
  186. if "Connected" in self._AList:
  187. if self._AList["Connected"] == 1:
  188. self._FootMsg[1] = "Disconnect"
  189. else:
  190. self._FootMsg[1] = ""
  191. self.GenList()
  192. def OnReturnBackCb(self):
  193. self.ReturnToUpLevelPage()
  194. self._Screen.Draw()
  195. self._Screen.SwapAndShow()
  196. def KeyDown(self,event):
  197. if event.key == CurKeys["A"] or event.key == CurKeys["Menu"]:
  198. self.ReturnToUpLevelPage()
  199. self._Screen.Draw()
  200. self._Screen.SwapAndShow()
  201. if event.key == CurKeys["Up"]:
  202. self.ScrollUp()
  203. self._Screen.Draw()
  204. self._Screen.SwapAndShow()
  205. if event.key == CurKeys["Down"]:
  206. self.ScrollDown()
  207. self._Screen.Draw()
  208. self._Screen.SwapAndShow()
  209. if event.key == CurKeys["Enter"]:
  210. self.Click()
  211. if event.key == CurKeys["X"]:
  212. self.TryToDisconnect()
  213. if event.key == CurKeys["Y"]:
  214. self.TryToForget()
  215. def Draw(self):
  216. if len(self._MyList) == 0:
  217. return
  218. self.ClearCanvas()
  219. if len(self._MyList) * InfoPageListItem._Height > self._Height:
  220. self._Ps._Width = self._Width - 10
  221. self._Ps._PosX = 9
  222. self._Ps.Draw()
  223. for i in self._MyList:
  224. i.Draw()
  225. self._Scroller.UpdateSize(len(self._MyList)*InfoPageListItem._Height, self._PsIndex*InfoPageListItem._Height)
  226. self._Scroller.Draw()
  227. else:
  228. self._Ps._Width = self._Width
  229. self._Ps.Draw()
  230. for i in self._MyList:
  231. i.Draw()
  232. class BleListSelector(PageSelector):
  233. _BackgroundColor = MySkinManager.GiveColor('Front')
  234. def __init__(self):
  235. self._PosX = 0
  236. self._PosY = 0
  237. self._Height = 0
  238. def AnimateDraw(self,x2,y2):
  239. pass
  240. def Draw(self):
  241. idx = self._Parent._PsIndex
  242. if idx < len( self._Parent._MyList):
  243. x = self._Parent._MyList[idx]._PosX+2
  244. y = self._Parent._MyList[idx]._PosY+1
  245. h = self._Parent._MyList[idx]._Height -3
  246. self._PosX = x
  247. self._PosY = y
  248. self._Height = h
  249. aa_round_rect(self._Parent._CanvasHWND,
  250. (x,y,self._Width-4,h),self._BackgroundColor,4,0,self._BackgroundColor)
  251. class BleListMessageBox(Label):
  252. _Parent = None
  253. def Draw(self):
  254. my_text = self._FontObj.render( self._Text,True,self._Color)
  255. w = my_text.get_width()
  256. h = my_text.get_height()
  257. x = (self._Parent._Width - w)/2
  258. y = (self._Parent._Height - h)/2
  259. padding = 10
  260. pygame.draw.rect(self._CanvasHWND,MySkinManager.GiveColor('White'),(x-padding,y-padding, w+padding*2,h+padding*2))
  261. pygame.draw.rect(self._CanvasHWND,MySkinManager.GiveColor('Black'),(x-padding,y-padding, w+padding*2,h+padding*2),1)
  262. self._CanvasHWND.blit(my_text,(x,y,w,h))
  263. class BluetoothPage(Page):
  264. _MyList = []
  265. #Wicd dbus part
  266. _Adapter = None
  267. _Dbus = None
  268. _Devices = None
  269. _BlePassword = ""
  270. _Connecting = False
  271. _Scanning = False
  272. _PrevState = None
  273. _Selector = None
  274. _ShowingMessageBox = False
  275. _MsgBox = None
  276. _ConnectTry = 0
  277. _BlockCb = None
  278. _LastStatusMsg = ""
  279. _FootMsg = ["Nav","Scan","Info","Back","TryConnect"]
  280. _Scroller = None
  281. _ListFontObj = fonts["notosanscjk15"]
  282. _InfoPage = None
  283. _ADAPTER_DEV = "hci0"
  284. _Offline = False
  285. def __init__(self):
  286. Page.__init__(self)
  287. self._MyList = []
  288. self._CanvasHWND = None
  289. def ShowBox(self,msg):
  290. self._MsgBox._Text = msg
  291. self._ShowingMessageBox = True
  292. self._Screen.Draw()
  293. self._MsgBox.Draw()
  294. self._Screen.SwapAndShow()
  295. def HideBox(self):
  296. self.Draw()
  297. self._ShowingMessageBox = False
  298. self._Screen.SwapAndShow()
  299. def Init(self):
  300. self._PosX = self._Index * self._Screen._Width
  301. self._Width = self._Screen._Width
  302. self._Height = self._Screen._Height
  303. #self._CanvasHWND = pygame.Surface((self._Width,self._Height))
  304. self._CanvasHWND = self._Screen._CanvasHWND
  305. ps = BleListSelector()
  306. ps._Parent = self
  307. ps._Width = Width - 12
  308. self._Ps = ps
  309. self._PsIndex = 0
  310. msgbox = BleListMessageBox()
  311. msgbox._CanvasHWND = self._CanvasHWND
  312. msgbox.Init(" ",MyLangManager.TrFont("veramono12"))
  313. msgbox._Parent = self
  314. self._MsgBox = msgbox
  315. self._Scroller = ListScroller()
  316. self._Scroller._Parent = self
  317. self._Scroller._PosX = 2
  318. self._Scroller._PosY = 2
  319. self._Scroller.Init()
  320. self.GenNetworkList()
  321. self._InfoPage = BleInfoPage()
  322. self._InfoPage._Screen = self._Screen
  323. self._InfoPage._Name = "BluetoothInfo"
  324. self._InfoPage.Init()
  325. def print_normal(self,address, properties):
  326. print("[ " + address + " ]")
  327. for key in properties.keys():
  328. value = properties[key]
  329. if type(value) is dbus.String:
  330. value = unicode(value).encode('ascii', 'replace')
  331. if (key == "Class"):
  332. print(" %s = 0x%06x" % (key, value))
  333. else:
  334. print(" %s = %s" % (key, value))
  335. print()
  336. properties["Logged"] = True
  337. def skip_dev(self,old_dev, new_dev):
  338. if not "Logged" in old_dev:
  339. return False
  340. if "Name" in old_dev:
  341. return True
  342. if not "Name" in new_dev:
  343. return True
  344. return False
  345. def DbusPropertiesChanged(self, interface, changed, invalidated, path):
  346. global devices
  347. print("DbusPropertiesChanged")
  348. if interface != "org.bluez.Device1":
  349. return
  350. if path in devices:
  351. dev = devices[path]
  352. if self.skip_dev(dev, changed):
  353. return
  354. devices[path] = dict(devices[path].items() + changed.items())
  355. else:
  356. devices[path] = changed
  357. if "Address" in devices[path]:
  358. address = devices[path]["Address"]
  359. else:
  360. address = "<unknown>"
  361. self._Devices = devices
  362. self.print_normal(address, devices[path])
  363. self.RefreshDevices()
  364. self.GenNetworkList()
  365. self._Screen.Draw()
  366. self._Screen.SwapAndShow()
  367. def ShutDownConnecting(self):
  368. print("Shutdownconnecting...")
  369. def AbortedAndReturnToUpLevel(self):
  370. self.HideBox()
  371. self._Screen._FootBar.ResetNavText()
  372. self.ReturnToUpLevelPage()
  373. self._Screen.Draw()
  374. self._Screen.SwapAndShow()
  375. def CheckIfBluetoothConnecting(self):
  376. return True
  377. def TryConnect(self):
  378. global bus
  379. if self._PsIndex >= len(self._MyList):
  380. return
  381. cur_li = self._MyList[self._PsIndex]
  382. print(cur_li._Path)
  383. if "Connected" in cur_li._Atts:
  384. if cur_li._Atts["Connected"] == 1:
  385. return
  386. proxy_obj = bus.get_object("org.bluez", cur_li._Path)
  387. dev = dbus.Interface(proxy_obj, "org.bluez.Device1")
  388. self._Screen._FootBar.UpdateNavText("Connecting")
  389. self.ShowBox(MyLangManager.Tr("Connecting"))
  390. try:
  391. dev.Connect()
  392. except Exception,e:
  393. print(str(e))
  394. self.HideBox()
  395. self._Screen._FootBar.ResetNavText()
  396. def RefreshDevices(self):
  397. global devices
  398. devices = {}
  399. try:
  400. proxy_obj = bus.get_object("org.bluez", "/")
  401. manager = dbus.Interface(proxy_obj,"org.freedesktop.DBus.ObjectManager")
  402. objects = manager.GetManagedObjects()
  403. for path, interfaces in objects.iteritems():
  404. if "org.bluez.Device1" in interfaces:
  405. devices[path] = interfaces["org.bluez.Device1"] ## like /org/bluez/hci0/dev_xx_xx_xx_yy_yy_yy
  406. except Exception,e:
  407. print(str(e))
  408. devices={}
  409. self._Devices = devices
  410. def GenNetworkList(self):
  411. self._MyList = []
  412. start_x = 0
  413. start_y = 0
  414. counter = 0
  415. for i,v in enumerate(self._Devices):
  416. if "Name" in self._Devices[v]:
  417. if len(self._Devices[v]["Name"]) < 2:
  418. continue
  419. if re.match("[0-9a-f]{2}([-:]?)[0-9a-f]{2}(\\1[0-9a-f]{2}){4}$", self._Devices[v]["Name"].lower()): ## skip xx:xx:xx:xx:xx
  420. continue
  421. else:
  422. continue
  423. ni = NetItem()
  424. ni._Parent = self
  425. ni._PosX = start_x
  426. ni._PosY = start_y + counter* NetItem._Height
  427. ni._Width = Width
  428. ni._FontObj = self._ListFontObj
  429. ni.Init(v,self._Devices[v])
  430. counter += 1
  431. self._MyList.append(ni)
  432. self._PsIndex = 0
  433. def Rescan(self):
  434. if self._Screen._CurrentPage != self:
  435. return
  436. self._Scanning = True
  437. self.ShowBox(MyLangManager.Tr("BluetoothScanning"))
  438. self._Screen._FootBar.UpdateNavText("Scanning")
  439. proxy_obj = self._Dbus.get_object("org.bluez", "/org/bluez/" + self._ADAPTER_DEV)
  440. adapter_props = dbus.Interface(proxy_obj,"org.freedesktop.DBus.Properties")
  441. discoverying = adapter_props.Get("org.bluez.Adapter1", "Discovering")
  442. print(discoverying)
  443. if self._Adapter!= None:
  444. try:
  445. self._Adapter.StopDiscovery()
  446. except Exception,e:
  447. print(str(e))
  448. try:
  449. self._Adapter.StartDiscovery()
  450. except Exception,e:
  451. print(str(e))
  452. def OnReturnBackCb(self):
  453. self.RefreshDevices()
  454. self.GenNetworkList()
  455. def OnLoadCb(self):
  456. self._Offline = False
  457. if self._Screen._TitleBar._InAirPlaneMode == False:
  458. out = commands.getstatusoutput("hcitool dev | grep hci0 |cut -f3") ## bluetooth maybe dead after airplane mode
  459. if len(out[1]) < 17:
  460. self._Offline = True
  461. print("Bluetooth OnLoadCb ,can not find hci0 alive,try to reboot")
  462. else:
  463. self.RefreshDevices()
  464. self.GenNetworkList()
  465. else:
  466. self._Offline = True
  467. def KeyDown(self,event):
  468. if event.key == CurKeys["A"] or event.key == CurKeys["Menu"]:
  469. if self._Offline == True:
  470. self.AbortedAndReturnToUpLevel()
  471. return
  472. if self._Adapter != None:
  473. try:
  474. self._Adapter.StopDiscovery()
  475. except Exception,e:
  476. print(str(e))
  477. _connecting = self.CheckIfBluetoothConnecting()
  478. if _connecting:
  479. self.ShutDownConnecting()
  480. self.ShowBox(MyLangManager.Tr("ShutDownConnecting"))
  481. self.AbortedAndReturnToUpLevel()
  482. else:
  483. self.AbortedAndReturnToUpLevel()
  484. else:
  485. self.HideBox()
  486. self.ReturnToUpLevelPage()
  487. self._Screen.Draw()
  488. self._Screen.SwapAndShow()
  489. self._Screen._FootBar.ResetNavText()
  490. if event.key == CurKeys["Up"]:
  491. self.ScrollUp()
  492. self._Screen.Draw()
  493. self._Screen.SwapAndShow()
  494. if event.key == CurKeys["Down"]:
  495. self.ScrollDown()
  496. self._Screen.Draw()
  497. self._Screen.SwapAndShow()
  498. if event.key == CurKeys["X"]:
  499. if self._Offline == False:
  500. self.Rescan()
  501. if event.key == CurKeys["Y"]:
  502. if len(self._MyList) == 0:
  503. return
  504. if self._Offline == True:
  505. return
  506. self._InfoPage._AList = self._MyList[self._PsIndex]._Atts
  507. self._InfoPage._Path = self._MyList[self._PsIndex]._Path
  508. self._Screen.PushPage(self._InfoPage)
  509. self._Screen.Draw()
  510. self._Screen.SwapAndShow()
  511. if event.key == CurKeys["B"]:
  512. if self._Offline == False:
  513. self.TryConnect()
  514. def Draw(self):
  515. self.ClearCanvas()
  516. if len(self._MyList) == 0:
  517. return
  518. if len(self._MyList) * NetItem._Height > self._Height:
  519. self._Ps._Width = self._Width - 11
  520. self._Ps.Draw()
  521. for i in self._MyList:
  522. i.Draw()
  523. self._Scroller.UpdateSize( len(self._MyList)*NetItem._Height, self._PsIndex*NetItem._Height)
  524. self._Scroller.Draw()
  525. else:
  526. self._Ps._Width = self._Width
  527. self._Ps.Draw()
  528. for i in self._MyList:
  529. i.Draw()
  530. class APIOBJ(object):
  531. _Page = None
  532. def __init__(self):
  533. pass
  534. def Init(self,main_screen):
  535. global bus,devices,adapter
  536. self._Page = BluetoothPage()
  537. self._Page._Dbus = bus
  538. self._Page._Devices = devices
  539. self._Page._Adapter = adapter
  540. self._Page._Screen = main_screen
  541. self._Page._Name ="Bluetooth"
  542. self._Page.Init()
  543. bus.add_signal_receiver(self._Page.DbusPropertiesChanged,
  544. dbus_interface = "org.freedesktop.DBus.Properties",
  545. signal_name = "PropertiesChanged",
  546. arg0 = "org.bluez.Device1",
  547. path_keyword = "path")
  548. def API(self,main_screen):
  549. if main_screen !=None:
  550. main_screen.PushPage(self._Page)
  551. main_screen.Draw()
  552. main_screen.SwapAndShow()
  553. OBJ = APIOBJ()
  554. def Init(main_screen):
  555. OBJ.Init(main_screen)
  556. def API(main_screen):
  557. OBJ.API(main_screen)