misc: small fix or general refactoring i did not bother commenting

This commit is contained in:
Simone Margaritelli 2019-10-24 13:09:54 +02:00
parent 032e183ff7
commit 6d70a24aae
No known key found for this signature in database
GPG Key ID: 82E42E7F3B34C97E

View File

@ -2,6 +2,7 @@ import _thread
from threading import Lock from threading import Lock
import time import time
import logging import logging
import random
from PIL import ImageDraw from PIL import ImageDraw
import pwnagotchi.utils as utils import pwnagotchi.utils as utils
@ -202,7 +203,18 @@ class View(object):
self.update() self.update()
def on_new_peer(self, peer): def on_new_peer(self, peer):
self.set('face', faces.FRIEND if peer.is_good_friend(self._config) else faces.EXCITED) face = ''
# first time they met, neutral mood
if peer.first_encounter():
face = random.choice((faces.AWAKE, faces.COOL))
# a good friend, positive expression
elif peer.is_good_friend():
face = random.choice((faces.MOTIVATED, faces.FRIEND, faces.HAPPY))
# normal friend, neutral-positive
else:
face = random.choice((faces.EXCITED, faces.HAPPY))
self.set('face', face)
self.set('status', self._voice.on_new_peer(peer)) self.set('status', self._voice.on_new_peer(peer))
self.update() self.update()
time.sleep(3) time.sleep(3)