def testLocation(self):
"""
Test to request location and validate coordinates
"""
coordinates = [12345, 12345]
result = Fona.GetLocation()
self.assertEqual(result, coordinates)
print("Location passed")
def __init__(self, menufile="ressources/menu.xml"):
# Initialisation du OLED
self.disp = SSD1306(rst="J4.12", dc="J4.14", cs="J4.11")
self.disp.begin()
self.clear_display()
self.image = Image.new('1', (self.disp.width, self.disp.height))
self.draw = ImageDraw.Draw(self.image)
# Initialisation du menu
self.maxlines = 6
self.menufile = menufile
self.init_menu()
# Initialisation du clavier
self.init_keypad()
# Initialisation du Fona
self.fona = Fona()
self.font = ImageFont.truetype('ressources/Minecraftia-Regular.ttf', 8)
# self.font = ImageFont.truetype('ressources/VeraMono.ttf', 8)
# Initialisation du Wifi
self.wifi = Wifi()
# Initialisation de la barre des tâches
self.tskbr_size = 128, 6
self.tskbr_padding = 2
self.tskbr_batt = True
self.tskbr_date = True
self.tskbr_time = True
self.tskbr_wifi = True
self.tskbr_fona = True
self.tskbr_message = True
self.tskbr_call = False # Ne fonctionne pas
def testReset(self):
"""
Test reset power sequence
"""
self.assert_(Fona.FonaReset())
class Phone:
"""Classe principale de l'application de téléphone cellulaire.
"""
# Initialisation
#================
def __init__(self, menufile="ressources/menu.xml"):
# Initialisation du OLED
self.disp = SSD1306(rst="J4.12", dc="J4.14", cs="J4.11")
self.disp.begin()
self.clear_display()
self.image = Image.new('1', (self.disp.width, self.disp.height))
self.draw = ImageDraw.Draw(self.image)
# Initialisation du menu
self.maxlines = 6
self.menufile = menufile
self.init_menu()
# Initialisation du clavier
self.init_keypad()
# Initialisation du Fona
self.fona = Fona()
self.font = ImageFont.truetype('ressources/Minecraftia-Regular.ttf', 8)
# self.font = ImageFont.truetype('ressources/VeraMono.ttf', 8)
# Initialisation du Wifi
self.wifi = Wifi()
# Initialisation de la barre des tâches
self.tskbr_size = 128, 6
self.tskbr_padding = 2
self.tskbr_batt = True
self.tskbr_date = True
self.tskbr_time = True
self.tskbr_wifi = True
self.tskbr_fona = True
self.tskbr_message = True
self.tskbr_call = False # Ne fonctionne pas
# Général
#=========
def init_keypad(self):
self.keypad_parent_conn, self.keypad_child_conn = Pipe()
self.keypad_sub = Process(target=keys.loop, args=(self.keypad_child_conn, ))
self.keypad_sub.start()
def clear_display(self):
self.disp.clear()
self.disp.display()
def clear_image(self):
self.draw.rectangle((0, 0, self.disp.width, self.disp.height), outline=0, fill=0)
def popup(self, message, padding=3, text_width=24):
width, height = self.image.size
message = wrap(message, width=text_width)
largest = 0
highest = 0
for l in message:
w = self.font.getsize(l)
if w[0] > largest:
largest = w[0]
if w[1] > highest:
highest = w[1]
self.draw.rectangle(((width - largest)/2 - padding,
(height - len(message)*highest)/2 - padding,
(width + largest)/2 + padding,
(height + len(message)*highest)/2 + padding,
),
outline=255,
fill=0)
line = 0
for l in message:
self.draw.text(((width - largest)/2,
(height - len(message)*highest)/2 + line*highest,
),
l,
font=self.font,
fill=255)
line += 1
self.disp.image(self.image)
self.disp.display()
# Actions (items Exec dans le menu)
#===================================
def shutdown(self, soft=False, restart=False):
"""Sortie gracieuse du programme, ou fermeture du Arietta.
"""
self.keypad_sub.terminate()
self.fona.turn_off()
self.clear_display()
if soft:
logging.info("Fin du programme.")
sys.exit()
elif not restart:
logging.info("Fermeture du Arietta.")
sub.call(["shutdown", "-h", "-H", "now"])
else:
logging.info("Redémarrage du Arietta.")
sub.call(["shutdown", "-r", "now"])
def shell(self, command):
try:
output = sub.check_output(command, shell=True)
except sub.CalledProcessError:
output = u"Erreur de l'exécution de : {}".format(command)
logging.error(u"Erreur de l'exécution de : {}".format(command))
if not len(output):
output = u"Ok"
return self.show(output)
def method(self, command):
output = eval(command)
return self.show(output)
# Accueil
#=========
def home(self):
self.clear_image()
# Indique la date / heure en haut à gauche
date = datetime.strftime(datetime.now(), "%Y-%m-%d %H:%M:%S")
self.draw.text((0, 0), date, font=self.font, fill=255)
# Affiche la barre des tâches
self.image.paste(self.get_tskbr_image(), (0, self.image.size[1] - self.tskbr_size[1]))
# Display image.
self.disp.image(self.image)
self.disp.display()
# Menus
#=======
def init_menu(self):
"""Initialise le menu.
"""
self.tree = etree.parse(self.menufile)
self.menu = self.tree.getroot()
self.buff = list()
self.cursor = [0]
for m in self.menu:
self.buff.append(m.find('Title').text)
def create_submenus(self, generator):
"""Cette fonction crée des sous-menus à partir d'une liste de tuples bâtie ainsi :
[('Nom', 'Titre', 'Action', 'Commande'), ...] où :
Nom = Nom de l'élément (interne)
Titre = Titre de l'élément du menu à afficher
Action = "Exec" ou "Generator" ou None
Commande = Nom de la fonction à appeler
Ce sont les générateurs qui produisent ces listes.
"""
# Génère les sous-menus à partir du générateur
logging.debug(u'générateur = {}'.format(generator))
try:
submenus = eval(u'self.{}'.format(generator))
except AttributeError:
logging.error("Méthode inexistante : Phone.{}".format(generator))
return [("Vide", u"Méthode inexistante", None, None)]
#logging.debug(u'submenus = {}'.format(submenus))
return submenus
def insert_submenus(self, submenus):
"""Insère les sous-menus dans l'arborescence existante.
"""
# Effacer les sous-menus actuels, si existants (niveau 1 et 2)
if self.menu[self.cursor[-1]].find('Submenu') is not None:
etree.strip_elements(self.menu[self.cursor[-1]], 'Submenu')
# Popule le nouveau sous-menu (niveau 1)
etree.SubElement(self.menu[self.cursor[-1]], 'Submenu')
for menu in submenus:
# Créé le sous-menu (niveau 2)
#logging.debug(u'menu = ({}, {}, {}, {})'.format(menu[0], menu[1], menu[2], menu[3]))
etree.SubElement(self.menu[self.cursor[-1]].find('Submenu'), menu[0])
# Nomme le sous-menu (niveau 2)
if menu[1] is not None:
etree.SubElement(self.menu[self.cursor[-1]].find('Submenu').find(menu[0]), 'Title')
try:
self.menu[self.cursor[-1]].find('Submenu').find(menu[0]).find('Title').text = menu[1]
except ValueError:
# All strings must be XML compatible: Unicode or ASCII, no NULL bytes or control characters
self.menu[self.cursor[-1]].find('Submenu').find(menu[0]).find('Title').text = u"Caractère(s) invalide(s)"
# Affecte l'action et la commande du sous-menu (niveau 2)
if menu[2] is not None and menu[3] is not None:
etree.SubElement(self.menu[self.cursor[-1]].find('Submenu').find(menu[0]), menu[2])
self.menu[self.cursor[-1]].find('Submenu').find(menu[0]).find(menu[2]).text = menu[3]
def update_buffer(self):
"""Mise à jour du buffer de menu pour l'affichage.
"""
self.buff = list()
self.cursor.append(0)
for m in self.menu:
try:
self.buff.append(unicode(m.find('Title').text))
except AttributeError:
logging.error('Aucun champ Title pour {}'.format(m.tag))
self.buff.append("")
def go_child(self):
"""Descend d'un niveau dans la navigation
"""
# Le prochain niveau est un générateur
if self.menu[self.cursor[-1]].find('Generator') is not None:
logging.debug('Génération de sous-menus dans {}'.format(self.menu[self.cursor[-1]].find('Title').text))
self.insert_submenus(self.create_submenus(self.menu[self.cursor[-1]].find('Generator').text))
self.menu = self.menu[self.cursor[-1]].find('Submenu')
self.update_buffer()
self.refresh_display()
# Le prochain niveau est une commande
elif self.menu[self.cursor[-1]].find('Exec') is not None:
logging.debug('Exécution de : {}'.format(self.menu[self.cursor[-1]].find('Exec').text))
try:
self.insert_submenus(eval(u'self.{}'.format(self.menu[self.cursor[-1]].find('Exec').text)))
except AttributeError:
logging.error("Méthode inexistante : Phone.{}".format(self.menu[self.cursor[-1]].find('Exec').text))
self.insert_submenus([("Vide", u"Méthode inexistante", None, None)])
self.menu = self.menu[self.cursor[-1]].find('Submenu')
self.update_buffer()
self.refresh_display()
# Le prochain niveau est un sous-menu
elif self.menu[self.cursor[-1]].find('Submenu') is not None:
logging.debug('Descente dans le sous-menu')
self.menu = self.menu[self.cursor[-1]].find('Submenu')
self.update_buffer()
self.refresh_display()
# Le prochain niveau n'a aucune action de définie (ex.: texte)
else:
logging.debug('Aucune action de définie pour {}'.format(self.menu[self.cursor[-1]].tag, ))
def go_parent(self):
"""Remonte d'un niveau dans la navigation.
"""
try:
self.menu = self.menu.find('..').find('..')
self.buff = list()
self.cursor.pop()
for m in self.menu:
self.buff.append(m.find('Title').text)
self.refresh_display()
except AttributeError:
logging.debug('Aucun menu parent pour {}'.format(self.menu.tag))
def refresh_display(self):
"""Synchronise le OLED avec le buffer
"""
self.clear_image()
i = 0
for l in range(self.cursor[-1], len(self.buff)):
if i < self.maxlines:
self.draw.text((0, 10*i), unicode(self.buff[l]), font=self.font, fill=255)
# if i == 0:
# draw.text((0, 10*i), u'> {}'.format(self.buff[l]), font=self.font, fill=255)
# else:
# draw.text((0, 10*i), unicode(self.buff[l]), font=self.font, fill=255)
i += 1
# Display image.
self.disp.image(self.image)
self.disp.display()
def show(self, message):
"""Retourne un menu composé des lignes de texte de "message".
"""
if message.__class__ != unicode:
message = message.decode('utf-8')
#logging.debug(u'message = {}'.format(message))
lines = message.splitlines()
split_msg = []
current_line = 0
space_width = self.font.getsize(' ')[0]
# Pour chaque ligne du message...
for line in lines:
words = line.split()
split_msg.append(u'')
line_width = 0
# Pour chaque mot du message...
for word in words:
word_width = self.font.getsize(word)[0]
# Si le mot entre dans la ligne actuelle...
if line_width + word_width <= self.disp.width:
split_msg[current_line] += word
line_width += word_width
# Sinon, si le mot n'entre pas dans un ligne vide...
elif word_width > self.disp.width:
current_line += 1
split_msg.append(u'')
line_width = 0
# Pour chaque caractère du mot...
for car in word:
car_width = self.font.getsize(car)[0]
# Si le caractère entre dans la ligne actuelle...
if line_width + car_width <= self.disp.width:
split_msg[current_line] += car
line_width += car_width
# Si le caractère n'entre pas dans la ligne actuelle...
else:
current_line += 1
split_msg.append(u'')
split_msg[current_line] += car
line_width = car_width
# Si le mot entre dans une nouvelle ligne vide...
else:
current_line += 1
split_msg.append(u'')
split_msg[current_line] += word
line_width = word_width
# Ajoute un espace après chaque mot.
if line_width + space_width <= self.disp.width:
split_msg[current_line] += u' '
line_width += space_width
# Nouvelle ligne (tel que message original)
current_line += 1
menu = list()
for line in split_msg:
menu.append((u'line' + unicode(len(menu)), line, None, None))
return menu
def scroll_down(self):
if self.cursor[-1] < len(self.buff) - 1:
self.cursor[-1] += 1
self.refresh_display()
else:
self.cursor[-1] = 0
self.refresh_display()
def scroll_up(self):
if self.cursor[-1] > 0:
self.cursor[-1] -= 1
self.refresh_display()
else:
self.cursor[-1] = len(self.buff) - 1
self.refresh_display()
def draw_text(self, text):
"""Affiche le texte pour le mode composition.
"""
self.clear_image()
self.draw.text((0, 10*0), unicode(text) + u"_", font=self.font, fill=255)
# Display image.
self.disp.image(self.image)
self.disp.display()
# Barre de notification
#=======================
def draw_batt(self, image, offset=0):
batt_size = 13, 6
draw = ImageDraw.Draw(image)
width, height = image.size
width -= offset
offset += batt_size[0] + self.tskbr_padding
m = re_battery.search(self.fona.get_battery())
if m:
charge = float(m.group(1))/100
else:
logging.error("Charge de batterie inconnue (pas de réponse du Fona).")
charge = 0.0
# Enveloppe batterie
draw.rectangle((width - (batt_size[0] - 1), 0, width - 1, batt_size[1] - 1),
outline=255,
fill=0)
draw.rectangle((width-batt_size[0], 1, width-batt_size[0], batt_size[1] - 2),
outline=255,
fill=0)
# Niveau de charge
draw.rectangle((width-(batt_size[0] - 2), 1, width-(batt_size[0] - 2) + charge*(batt_size[0] - 3), batt_size[1] - 2),
outline=255,
fill=255)
return image, offset
def get_date(self):
return datetime.strftime(datetime.now(), "%y%m%d")
def get_time(self):
return datetime.strftime(datetime.now(), "%H:%M")
def draw_fona(self, image, offset=0, status=2): # Corriger status
icon_size = 8, 6
draw = ImageDraw.Draw(image)
width, height = image.size
width -= offset
offset += icon_size[0] + self.tskbr_padding
# 2 = Connected
if self.fona.network_status.get():
status = 2
# 1 = On
elif self.fona.power_status.get():
status = 1
# 0 = Off
else:
status = 0
rect = (width - icon_size[0], 4, width - icon_size[0] + 1, 5)
draw.rectangle(rect, outline=255, fill=0)
if status > 0:
rect = (width - icon_size[0] + 3, 2, width - icon_size[0] + 4, 5)
draw.rectangle(rect, outline=255, fill=0)
if status > 1:
rect = (width - icon_size[0] + 6, 0, width - icon_size[0] + 7, 5)
draw.rectangle(rect, outline=255, fill=0)
return image, offset
def draw_message(self, image, offset=0):
status = self.fona.new_sms()
if not status:
return image, offset
icon_size = 10, 6
draw = ImageDraw.Draw(image)
width, height = image.size
width -= offset
offset += icon_size[0] + self.tskbr_padding
lines = ((width - icon_size[0] + 0, 0, width - icon_size[0] + 9, 0),
(width - icon_size[0] + 2, 1, width - icon_size[0] + 7, 1),
(width - icon_size[0] + 4, 2, width - icon_size[0] + 5, 2),
(width - icon_size[0] + 0, 2, width - icon_size[0] + 1, 2),
(width - icon_size[0] + 8, 2, width - icon_size[0] + 9, 2),
(width - icon_size[0] + 0, 3, width - icon_size[0] + 3, 3),
(width - icon_size[0] + 6, 3, width - icon_size[0] + 9, 3),
(width - icon_size[0] + 0, 4, width - icon_size[0] + 9, 4),
(width - icon_size[0] + 0, 5, width - icon_size[0] + 9, 5),
)
for l in lines:
draw.line(l, fill=255)
return image, offset
def draw_call(self, image, offset=0):
status = self.fona.ring.get()
if not status:
return image, offset
icon_size = 14, 6
draw = ImageDraw.Draw(image)
width, height = image.size
width -= offset
offset += icon_size[0] + self.tskbr_padding
lines = ((width - icon_size[0] + 4, 0, width - icon_size[0] + 9, 0),
(width - icon_size[0] + 2, 1, width - icon_size[0] + 11, 1),
(width - icon_size[0] + 1, 2, width - icon_size[0] + 12, 2),
(width - icon_size[0] + 0, 3, width - icon_size[0] + 3, 3),
(width - icon_size[0] + 10, 3, width - icon_size[0] + 13, 3),
(width - icon_size[0] + 0, 4, width - icon_size[0] + 3, 4),
(width - icon_size[0] + 10, 4, width - icon_size[0] + 13, 4),
(width - icon_size[0] + 0, 5, width - icon_size[0] + 3, 5),
(width - icon_size[0] + 10, 5, width - icon_size[0] + 13, 5),
)
for l in lines:
draw.line(l, fill=255)
return image, offset
def draw_wifi(self, image, offset=0):
self.wifi.get_status()
icon_size = 9, 6
draw = ImageDraw.Draw(image)
width, height = image.size
width -= offset
# Connecté
if self.wifi.on and self.wifi.essid is not None:
logging.debug("Wifi connecté à : {}".format(self.wifi.essid))
logging.debug("Qualité connexion : {}/70".format(self.wifi.quality))
# High signal quality
if self.wifi.quality > 35:
status = 2
else:
status = 1
# Déconnecté
elif self.wifi.on:
logging.debug("Wifi déconnecté.")
status = 0
# Désactivé
else:
return image, offset
p = [((4, 5), ),
((4, 5), (2, 4), (3, 3), (4, 3), (5, 3), (6, 4)),
((4, 5), (2, 4), (3, 3), (4, 3), (5, 3), (6, 4), (0, 2), (1, 1), (2, 1), (3, 0), (4, 0), (5, 0), (6, 1), (7, 1), (8, 2)),
]
# Décalage des points
points = tuple(map(lambda x: (width - icon_size[0] + x[0], x[1]), p[status]))
# Variables renvoyées
offset += icon_size[0] + self.tskbr_padding
draw.point(points, fill=255)
return image, offset
def get_tskbr_image(self):
image = Image.new('1', (self.tskbr_size[0], self.tskbr_size[1]))
offset = 0
if self.tskbr_batt:
image, offset = self.draw_batt(image, offset)
if self.tskbr_wifi:
image, offset = self.draw_wifi(image, offset)
if self.tskbr_fona:
image, offset = self.draw_fona(image, offset)
if self.tskbr_message:
image, offset = self.draw_message(image, offset)
if self.tskbr_call:
image, offset = self.draw_call(image, offset)
return image
