Clean up / ported new functions to daily-main

Prepped for pull into master

Author: cibear

MessageClass.py

@@ -5,6 +5,8 @@
 import locale
 from PIL._imaging import font
 from MessageClass import *
+import random
+import os
 
 ### error handling
 #error class that handles exceptions in pick_item routine
@@ -15,16 +17,26 @@ def __str__(self):
         return self.msg
 #########################
 
-
+fonts = ["ArialBd.ttf","Verdana.ttf","ELEPHNT.TTF"]
 
 # define usable fonts
 font = ImageFont.truetype('ArialBd.ttf', 14, encoding='unic')
 header_font = ImageFont.truetype('Verdana.ttf', 14, encoding='unic')
+font2 = ImageFont.truetype('ELEPHNT.ttf', 14, encoding='unic')
 
 # stores pictures usable for messages
-Pics = {}
-Pics["Rose"]="gfx/rose2.png"
-Pics["Schweinchen"]="gfx/Schweinchen.png"
+symbol_folder = os.path.dirname(os.path.realpath(__file__)) + "/symbols"
+Pics = []
+
+#populate Pics list with symbols
+for dirname, dirnames, filenames in os.walk(symbol_folder):
+    for filename in filenames:
+        if filename.split(".")[1] == "png":
+            Pics.append(os.path.join(dirname,filename))
+
+#Pics["Rose"]="gfx/rose2.png"
+#Pics["Schweinchen"]="gfx/Schweinchen.png"
+#Pics["Kalender"]="gfx/calendar.png"
 
 
 
@@ -39,8 +51,9 @@ def __str__(self):
 
 # Message Styles
 Message_Style = {
-    "Rose":[[msg_header,header_font], msg_spacer, ["png",Pics['Rose']], 30, "text", msg_spacer, [msg_bottom,header_font]],
-    "Schweinchen":[[msg_header,header_font], msg_spacer, ["png",Pics['Schweinchen']], 30, "text", msg_spacer, [msg_bottom,header_font]]
+#    "Rose":[[msg_header,header_font], msg_spacer, ["png",Pics['Rose']], 30, "text", msg_spacer, [msg_bottom,header_font]],
+#    "Schweinchen":[[msg_header,header_font], msg_spacer, ["png",Pics['Schweinchen']], 30, "text", msg_spacer, [msg_bottom,header_font]],
+    "Random":[[msg_header,header_font], msg_spacer, ["png",random.choice(Pics)], 30, "text", msg_spacer, [msg_bottom,header_font]]
 }
 
 

MessageClass.pyc

@@ -31,6 +31,13 @@
 lastId       = '1'   # State information passed to/from interval script
 printer      = Adafruit_Thermal("/dev/serial0", 19200, timeout=5)
 
+# set item path
+dir_path = os.path.dirname(os.path.realpath(__file__))
+item_folder = dir_path + '/items'
+
+# initialize item of the day
+picked_item = ""
+
 # Daytime to reset the printer to print the new message of the next day
 hrs  = 4
 mins = 0
@@ -39,7 +46,10 @@
 def tap():
   print("Button press detected.")
   GPIO.output(ledPin, GPIO.HIGH)  # LED on while working
-  subprocess.call(["python", "collected_before.py"])
+  #subprocess.call(["python", "collected_before.py"])
+  #print today's message again
+  if picked_item != "":
+      printer.printImage(png_from_item(picked_item), True)
   GPIO.output(ledPin, GPIO.LOW)
 
 
@@ -69,9 +79,20 @@ def interval():
 def daily():
   print("First button press of the day...")
   GPIO.output(ledPin, GPIO.HIGH)
-  subprocess.call(["python", "daily_message.py"])
+  #subprocess.call(["python", "daily_message.py"])
+  printer.printImage(Image.open(png_from_item(picked_item)), True)
   GPIO.output(ledPin, GPIO.LOW)
 
+# takes a file path and return ready-to-print png image, converting text files by invoking MessageClass
+def png_from_item(picked_item):
+    if picked_item.rsplit(".",1)[1] == "png":
+        print "Image printed..."
+        return Image.open(picked_item)
+    elif picked_item.rsplit(".",1)[1] == "txt":
+        textfile_handle = open(picked_item, "r")
+        textfile_output = "".join(textfile_handle.readlines())
+        random_font = ImageFont.truetype(random.choice(fonts), 14, encoding='unic')
+        return Message(textfile_output,random.choice(Message_Style.values())).BuildMessage(random_font)
 
 # Initialization
 
@@ -112,7 +133,8 @@ def daily():
 		exit(0)
 
 # Print greeting image.
-printer.print('Started at ' + datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S'))
+#printer.print('Started at ' + datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S'))
+printer.printImage(Image.open('gfx/helloworld.png'), True)
 printer.feed(3)
 GPIO.output(ledPin, GPIO.LOW)
 
@@ -163,6 +185,14 @@ def daily():
   l = time.localtime()
   if (60 * l.tm_hour + l.tm_min) > (60 * hrs + mins) and (60 * l.tm_hour + l.tm_min) < (60 * hrs + mins + 2):
     dailyFlag = False
+    #clear current item of the day by moving file to 'used' bin
+  ###if picked_item != "":
+  ###    move_to_bin(picked_item)    
+    # pick item of the day
+    picked_item = filewalker.pick_item(item_folder,date.today())
+    # print item of the day
+    printer.printImage(png_from_item(picked_item), True)
+    
 
   # LED blinks while idle, for a brief interval every 2 seconds.
   # Pin 18 is PWM-capable and a "sleep throb" would be nice, but

daily-main.py

@@ -19,27 +19,48 @@
 
 ###printer.setSize('L')   # Set type size, accepts 'S', 'M', 'L'
 #printer.println(filewalker.pick_random_pic(item_folder))
+
+
+# takes a file path and return ready-to-print png image, converting text files by invoking MessageClass
+def png_from_item(picked_item):
+    if picked_item.rsplit(".",1)[1] == "png":
+        print "Image printed..."
+        return Image.open(picked_item)
+    elif picked_item.rsplit(".",1)[1] == "txt":
+        textfile_handle = open(picked_item, "r")
+        textfile_output = "".join(textfile_handle.readlines())
+        random_font = ImageFont.truetype(random.choice(fonts), 14, encoding='unic')
+        return Message(textfile_output,random.choice(Message_Style.values())).BuildMessage(random_font)
+
+
+# choose a file
 picked_item = filewalker.pick_item(item_folder,date.today())
 
+#convert to png
+png_from_item(picked_item).show()
+
+
 #process according to file type
-print "picked item: " + picked_item
-print picked_item.rsplit(".",1)[1]
-print "png?" + str(picked_item.rsplit(".",1)[1] == "png")
-print "txt?" + str(picked_item.rsplit(".",1)[1] == "txt")
-
-if picked_item.rsplit(".",1)[1] == "png":
-    ###printer.printImage(Image.open(filewalker.pick_item(item_folder)), True)
-    print "Image printed..."
-elif picked_item.rsplit(".",1)[1] == "txt":
-    textfile_handle = open(picked_item, "r")
-    textfile_output = "".join(textfile_handle.readlines())
-    #print textfile_output
-    print textfile_output
-    XX = Message(textfile_output,random.choice(Message_Style.values()))
-    XX.print_me()
-    #XY = Message("This is the text! I see what you did there...",Message_Style["Schweinchen"])
-    XX.BuildMessage(font).show()
-    #returned_image = msg_block2png(Message, font, wrap_limit, offset_x, offset_y)
+# print "picked item: " + picked_item
+# print picked_item.rsplit(".",1)[1]
+# print "png?" + str(picked_item.rsplit(".",1)[1] == "png")
+# print "txt?" + str(picked_item.rsplit(".",1)[1] == "txt")
+
+# if picked_item.rsplit(".",1)[1] == "png":
+#     ###printer.printImage(Image.open(filewalker.pick_item(item_folder)), True)
+#     print "Image printed..."
+# elif picked_item.rsplit(".",1)[1] == "txt":
+#     textfile_handle = open(picked_item, "r")
+#     textfile_output = "".join(textfile_handle.readlines())
+#     #print textfile_output
+#     print textfile_output
+#     XX = Message(textfile_output,random.choice(Message_Style.values()))
+#     XX.print_me()
+#     #XY = Message("This is the text! I see what you did there...",Message_Style["Schweinchen"])
+#     random_font = ImageFont.truetype(random.choice(fonts), 14, encoding='unic')
+#     #XX.BuildMessage(random_font).show()
+#     Message(textfile_output,random.choice(Message_Style.values())).BuildMessage(random_font).show()
+#     #returned_image = msg_block2png(Message, font, wrap_limit, offset_x, offset_y)
 
     ###printer.println('Now printing text file:')
     ###printer.println(textfile_output)

daily_message_tester.py

@@ -0,0 +1,166 @@
+#!/usr/bin/python
+
+# Main script for Adafruit Internet of Things Printer 2.  Monitors button
+# for taps and holds, performs periodic actions (Twitter polling by default)
+# and daily actions (Sudoku and weather by default).
+# Written by Adafruit Industries.  MIT license.
+#
+# MUST BE RUN AS ROOTi (due to GPIO access)
+#
+# Required software includes Adafruit_Thermal, Python Imaging and PySerial
+# libraries. Other libraries used are part of stock Python install.
+#
+# Resources:
+# http://www.adafruit.com/products/597 Mini Thermal Receipt Printer
+# http://www.adafruit.com/products/600 Printer starter pack
+
+from __future__ import print_function
+import RPi.GPIO as GPIO
+import subprocess, time, Image, socket
+from Adafruit_Thermal import *
+
+ledPin       = 18
+buttonPin    = 23
+holdTime     = 2     # Duration for button hold (shutdown)
+tapTime      = 0.01  # Debounce time for button taps
+nextInterval = 0.0   # Time of next recurring operation
+dailyFlag    = False # Set after daily trigger occurs
+lastId       = '1'   # State information passed to/from interval script
+printer      = Adafruit_Thermal("/dev/serial0", 19200, timeout=5)
+
+
+# Called when button is briefly tapped.  Invokes time/temperature script.
+def tap():
+  GPIO.output(ledPin, GPIO.HIGH)  # LED on while working
+  subprocess.call(["python", "timetemp.py"])
+  GPIO.output(ledPin, GPIO.LOW)
+
+
+# Called when button is held down.  Prints image, invokes shutdown process.
+def hold():
+  GPIO.output(ledPin, GPIO.HIGH)
+  printer.printImage(Image.open('gfx/goodbye.png'), True)
+  printer.feed(3)
+  subprocess.call("sync")
+  subprocess.call(["shutdown", "-h", "now"])
+  GPIO.output(ledPin, GPIO.LOW)
+
+
+# Called at periodic intervals (30 seconds by default).
+# Invokes twitter script.
+def interval():
+  GPIO.output(ledPin, GPIO.HIGH)
+  p = subprocess.Popen(["python", "twitter.py", str(lastId)],
+    stdout=subprocess.PIPE)
+  GPIO.output(ledPin, GPIO.LOW)
+  return p.communicate()[0] # Script pipes back lastId, returned to main
+
+
+# Called once per day (6:30am by default).
+# Invokes weather forecast and sudoku-gfx scripts.
+def daily():
+  GPIO.output(ledPin, GPIO.HIGH)
+  subprocess.call(["python", "forecast.py"])
+  subprocess.call(["python", "sudoku-gfx.py"])
+  GPIO.output(ledPin, GPIO.LOW)
+
+
+# Initialization
+
+# Use Broadcom pin numbers (not Raspberry Pi pin numbers) for GPIO
+GPIO.setmode(GPIO.BCM)
+
+# Enable LED and button (w/pull-up on latter)
+GPIO.setup(ledPin, GPIO.OUT)
+GPIO.setup(buttonPin, GPIO.IN, pull_up_down=GPIO.PUD_UP)
+
+# LED on while working
+GPIO.output(ledPin, GPIO.HIGH)
+
+# Processor load is heavy at startup; wait a moment to avoid
+# stalling during greeting.
+time.sleep(30)
+
+# Show IP address (if network is available)
+try:
+	s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+	s.connect(('8.8.8.8', 0))
+	printer.print('My IP address is ' + s.getsockname()[0])
+	printer.feed(3)
+except:
+	printer.boldOn()
+	printer.println('Network is unreachable.')
+	printer.boldOff()
+	printer.print('Connect display and keyboard\n'
+	  'for network troubleshooting.')
+	printer.feed(3)
+	exit(0)
+
+# Print greeting image
+printer.printImage(Image.open('gfx/hello.png'), True)
+printer.feed(3)
+GPIO.output(ledPin, GPIO.LOW)
+
+# Poll initial button state and time
+prevButtonState = GPIO.input(buttonPin)
+prevTime        = time.time()
+tapEnable       = False
+holdEnable      = False
+
+# Main loop
+while(True):
+
+  # Poll current button state and time
+  buttonState = GPIO.input(buttonPin)
+  t           = time.time()
+
+  # Has button state changed?
+  if buttonState != prevButtonState:
+    prevButtonState = buttonState   # Yes, save new state/time
+    prevTime        = t
+  else:                             # Button state unchanged
+    if (t - prevTime) >= holdTime:  # Button held more than 'holdTime'?
+      # Yes it has.  Is the hold action as-yet untriggered?
+      if holdEnable == True:        # Yep!
+        hold()                      # Perform hold action (usu. shutdown)
+        holdEnable = False          # 1 shot...don't repeat hold action
+        tapEnable  = False          # Don't do tap action on release
+    elif (t - prevTime) >= tapTime: # Not holdTime.  tapTime elapsed?
+      # Yes.  Debounced press or release...
+      if buttonState == True:       # Button released?
+        if tapEnable == True:       # Ignore if prior hold()
+          tap()                     # Tap triggered (button released)
+          tapEnable  = False        # Disable tap and hold
+          holdEnable = False
+      else:                         # Button pressed
+        tapEnable  = True           # Enable tap and hold actions
+        holdEnable = True
+
+  # LED blinks while idle, for a brief interval every 2 seconds.
+  # Pin 18 is PWM-capable and a "sleep throb" would be nice, but
+  # the PWM-related library is a hassle for average users to install
+  # right now.  Might return to this later when it's more accessible.
+  if ((int(t) & 1) == 0) and ((t - int(t)) < 0.15):
+    GPIO.output(ledPin, GPIO.HIGH)
+  else:
+    GPIO.output(ledPin, GPIO.LOW)
+
+  # Once per day (currently set for 6:30am local time, or when script
+  # is first run, if after 6:30am), run forecast and sudoku scripts.
+  l = time.localtime()
+  if (60 * l.tm_hour + l.tm_min) > (60 * 6 + 30):
+    if dailyFlag == False:
+      daily()
+      dailyFlag = True
+  else:
+    dailyFlag = False  # Reset daily trigger
+
+  # Every 30 seconds, run Twitter scripts.  'lastId' is passed around
+  # to preserve state between invocations.  Probably simpler to do an
+  # import thing.
+  if t > nextInterval:
+    nextInterval = t + 30.0
+    result = interval()
+    if result is not None:
+      lastId = result.rstrip('\r\n')
+

gfx/calendar.png

@@ -62,6 +62,7 @@ def daily():
   GPIO.output(ledPin, GPIO.HIGH)
   subprocess.call(["python", "forecast.py"])
   subprocess.call(["python", "sudoku-gfx.py"])
+  
   GPIO.output(ledPin, GPIO.LOW)
 
 
@@ -97,7 +98,7 @@ def daily():
 	exit(0)
 
 # Print greeting image
-printer.printImage(Image.open('gfx/hello.png'), True)
+printer.printImage(Image.open('gfx/helloworld.png'), True)
 printer.feed(3)
 GPIO.output(ledPin, GPIO.LOW)