class PajaBot(SingleServerIRCBot):
def __init__(self):
config = ConfigParser.ConfigParser()
configfile = '/home/ovi/pajabot/bot.conf'
if (os.path.isfile('/home/ovi/pajabot/local.conf')):
configfile = '/home/ovi/pajabot/local.conf'
config.read(configfile)
self.server = config.get("bot","server")
self.ircchannel = config.get("bot","channel")
self.nick = config.get("bot","nick")
self.realname = config.get("bot","realname")
self.shoturl = config.get("bot","shoturl")
self.messageasaction = config.getboolean("bot","messageasaction")
self.vaasa = config.getboolean("bot","vaasa")
self.printer_ip = config.get("bot","printer")
try:
self.password = config.get("bot","password")
except ConfigParser.NoOptionError:
print "no password"
self.password = ''
try:
self.apikey = config.get("bot","apikey")
except ConfigParser.NoOptionError:
print "no apikey"
self.apikey = ''
try:
self.rss_url = config.get("vaasa","rss")
except ConfigParser.NoOptionError:
print "not in vaasa?"
self.rss_url = ''
self.rss_timestamp = ''
print "-- config --"
print self.server
print self.ircchannel
print self.nick
print self.realname
print self.messageasaction
print self.vaasa
print self.rss_url
print self.password
print self.printer_ip
print "-- end config --"
spec = ServerSpec(self.server)
SingleServerIRCBot.__init__(self, [spec], self.nick, self.realname)
self.reconnection_interval = 60
self.running = True
self.channel = self.ircchannel
self.doorStatus = None
self.spaceapi = SpaceAPI(config.get("bot", "spaceapiurl"))
self.camera = RPiCamera()
if (self.vaasa):
from iioo import IiOo
self.iioo = IiOo()
else:
self.iioo = RpiCamera()
#self.iioo = RPiCamera()
self.lightStatus = self.iioo.checkLights()
self.statusMessage = "Hello world"
def run(self):
spec = ServerSpec(self.server)
SingleServerIRCBot.__init__(self, [spec], self.nick, self.realname)
self._connect()
self.lightCheck = 0 # Check only every N loops
self.timestamp = datetime.datetime.now()
self.updateStatus()
feed_read_counter=99
mqttclient = paho.Client()
mqttclient.on_subscribe = on_subscribe
mqttclient.on_message = on_message
try:
mqttclient.connect("tunkki9", 1883)
except:
print "mqtt not connected"
mqttclient.loop_start()
mqttclient.subscribe("door/#", qos=1)
while(self.running):
self.checkLights()
self.mqtt_door()
# feed_read_counter +=1
# if (self.vaasa and feed_read_counter==100):
# feed_read_counter = 0
# self.read_feed()
# self.updateStatus()
try:
self.reactor.process_once(0.2)
except UnicodeDecodeError:
pass
# print 'Somebody said something in non-utf8'
# traceback.print_exc(file=sys.stdout)
except irc.client.ServerNotConnectedError:
print 'Not connected. Can not do anything atm.'
time.sleep(0.5)
def read_feed(self):
c = self.connection
global rss_timestamp
rssfeed = feedparser.parse(self.rss_url)
if len(rssfeed.entries)>0:
latest = rssfeed.entries[len(rssfeed.entries)-1]
if latest.id in self.rss_timestamp:
variable = 2
else:
self.rss_timestamp = latest.id
try:
self.say("door opened by " + latest.title)
print "new openings " + latest.title
except:
print "not connected"
def mqtt_door(self):
global newmqttmessage
global mqttmessage
# print newmqttmessage
# print mqttmessage
if newmqttmessage:
newmqttmessage = False
try:
self.say("door opened by " + mqttmessage)
print "new opening " + mqttmessage
except:
print "not connected"
mqttmessage = ""
def checkLights(self):
self.lightCheck -= 1
if self.lightCheck < 0:
# print 'Checking lights..'
newLights = self.iioo.checkLights()
if (newLights is None):
newLights = self.lightStatus
if newLights is not self.lightStatus:
# self.say("status " + str(self.lightStatus))
# self.say("newstatus " + str(newLights))
newTimestamp = datetime.datetime.now()
timeDelta = str(newTimestamp - self.timestamp).split('.')[0]
lss = 'lights ' + ('went off (lights were illuminated for ' if not newLights else 'on (darkness had fallen for ') + timeDelta + ')'
self.say(lss)
self.lightStatus = newLights
self.timestamp = newTimestamp
self.updateStatus()
self.lightCheck = 12
def say(self, text):
if self.messageasaction:
self.connection.action(self.channel, text)
else:
self.connection.privmsg(self.channel, text)
def updateStatus(self):
openstatus = ('true' if self.lightStatus else 'false')
self.statusMessage = ('The lab is manned' if self.lightStatus else 'No one here atm')
print 'Updating status: ' + openstatus + ', ' + self.statusMessage
self.spaceapi.updateStatus(openstatus, self.statusMessage)
# os.system('/home/pi/pajabot/scripts/updatestatus.sh ' + openstatus + ' "' + self.statusMessage + '"')
self.camera.takeShotCommand()
def on_welcome(self, c, e):
c.join(self.channel)
if (self.password!=''): c.privmsg("nickserv", "IDENTIFY " + self.password)
def sayDoorStatus(self):
c = self.connection
ds = self.doorStatus
dss = 'not giving status a.k.a. broken'
if ds is False:
dss = 'open'
if ds is True:
dss = 'closed'
dss = 'door is ' + dss
self.say(dss)
def sayPrinterStatus(self):
"""
Fetch and say 3D printer status using OctoPrint JSON API.
API documentation: http://docs.octoprint.org/en/master/api/job.html
"""
try:
req = urllib2.Request('http://' + self.printer_ip + '/api/job')
req.add_header('X-Api-Key', self.apikey)
response = urllib2.urlopen(req)
json_text = response.read()
response.close()
except urllib2.URLError, e:
self.say("Printer status: API checking failed: %s" % e.reason)
return
except urllib2.HTTPError, e:
self.say("Printer status: API checking failed: %s" % e.reason)
return
class PajaBot(SingleServerIRCBot):
def __init__(self):
config = ConfigParser.ConfigParser()
configfile = '~/pajabot/bot.conf'
if (os.path.isfile('~/pajabot/local.conf')):
configfile = '~/pajabot/local.conf'
if (os.path.isfile('local.conf')):
configfile = 'local.conf'
print "Reading config from " + configfile
config.read(configfile)
self.server = config.get("bot","server")
self.ircchannel = config.get("bot","channel")
self.nick = config.get("bot","nick")
self.realname = config.get("bot","realname")
self.shoturl = config.get("bot","shoturl")
self.messageasaction = config.getboolean("bot","messageasaction")
self.vaasa = config.getboolean("bot","vaasa")
self.printer_ip = config.get("bot","printer")
try:
self.password = config.get("bot","password")
except ConfigParser.NoOptionError:
print "no password"
self.password = ''
try:
self.rss_url = config.get("vaasa","rss")
except ConfigParser.NoOptionError:
print "not in vaasa?"
self.rss_url = ''
self.rss_timestamp = ''
print "-- config --"
print self.server
print self.ircchannel
print self.nick
print self.realname
print self.messageasaction
print self.vaasa
print self.rss_url
print self.password
print self.printer_ip
print "-- end config --"
self.reconnection_interval = 60
self.running = True
self.channel = self.ircchannel
self.doorStatus = None
self.spaceapi = SpaceAPI(config.get("bot", "spaceapiurl"))
self.camera = RPiCamera()
self.lightStatus = self.camera.checkLights()
self.statusMessage = "Hello world"
def run(self):
spec = ServerSpec(self.server)
SingleServerIRCBot.__init__(self, [spec], self.nick, self.realname)
self._connect()
self.lightCheck = 0 # Check only every N loops
self.timestamp = datetime.datetime.now()
self.updateStatus()
while(self.running):
self.checkLights()
if (self.vaasa): self.read_feed()
try:
self.reactor.process_once(0.2)
except UnicodeDecodeError:
pass
# print 'Somebody said something in non-utf8'
# traceback.print_exc(file=sys.stdout)
except irc.client.ServerNotConnectedError:
print 'Not connected. Can not do anything atm.'
time.sleep(0.5)
def read_feed(self):
c = self.connection
global rss_timestamp
rssfeed = feedparser.parse(self.rss_url)
if len(rssfeed.entries)>0:
latest = rssfeed.entries[len(rssfeed.entries)-1]
if latest.id in self.rss_timestamp:
variable = 2
else:
self.rss_timestamp = latest.id
try:
self.say("door opened by " + latest.title)
print "new openings " + latest.title
except:
print "not connected"
def checkLights(self):
self.lightCheck -= 1
if self.lightCheck < 0:
# print 'Checking lights..'
newLights = self.camera.checkLights()
if newLights is not self.lightStatus:
newTimestamp = datetime.datetime.now()
timeDelta = str(newTimestamp - self.timestamp).split('.')[0]
lss = 'lights ' + ('went off (lights were illuminated for ' if not newLights else 'on (darkness had fallen for ') + timeDelta + ')'
self.say(lss)
self.lightStatus = newLights
self.timestamp = newTimestamp
self.updateStatus()
self.lightCheck = 120
def say(self, text):
if self.messageasaction:
self.connection.action(self.channel, text)
else:
self.connection.privmsg(self.channel, text)
def updateStatus(self):
openstatus = ('true' if self.lightStatus else 'false')
self.statusMessage = ('The lab is manned' if self.lightStatus else 'No one here atm')
print 'Updating status: ' + openstatus + ', ' + self.statusMessage
self.spaceapi.updateStatus(openstatus, self.statusMessage)
# os.system('/home/pi/pajabot/scripts/updatestatus.sh ' + openstatus + ' "' + self.statusMessage + '"')
self.camera.takeShotCommand()
def on_welcome(self, c, e):
c.join(self.channel)
if (self.password!=''): c.privmsg("nickserv", "IDENTIFY " + self.password)
def sayDoorStatus(self):
c = self.connection
ds = self.doorStatus
dss = 'broken'
if ds is False:
dss = 'open'
if ds is True:
dss = 'closed'
dss = 'door is ' + dss
self.say(dss)
def on_nicknameinuse(self, c, e):
c.nick(c.get_nickname() + "_")
def on_pubmsg(self, c, e):
cmd = e.arguments[0].split()[0]
if cmd[0] == "!":
cmd = cmd[1:].upper()
if commands.has_key(cmd):
commands[cmd].index(self, c, e)
else:
cmd=e.arguments[0]
if cmd=='!kuole':
self.running = False
SingleServerIRCBot.die(self, 'By your command')
if (cmd=='!ovi') or (cmd=='!door'):
self.sayDoorStatus()
if (cmd=='!valot') or (cmd=='!lights'):
self.say('lights are ' + ('on' if self.lightStatus else 'off'))
if (cmd=='!checksum') or (cmd=='!checksum'):
self.say('pixelvar: ' + str(self.camera.checkSum()))
if (cmd=='!printer') or (cmd=='!tulostin'):
ping_response = subprocess.Popen(["/bin/ping", "-c1", "-w2", self.printer_ip], stdout=subprocess.PIPE).stdout.read()
if ('rtt' in ping_response):
self.say('printer is online')
else:
self.say('printer is offline')
print('p: ' + str(ping_response))
if cmd=='!shot':
self.camera.takeShotCommand()
c.privmsg(self.channel, self.shoturl + ('' if self.lightStatus else ' (pretty dark, eh)'))
if cmd=='!gitpull':
os.system('/home/pi/pajabot/scripts/gitpull.sh')
c.privmsg(self.channel, 'Pulled from git, restarting..')
self.restart_program()
if cmd=='!update':
self.updateStatus()
c.privmsg(self.channel, 'Done')
def _dispatcher(self, c, e):
eventtype = e.type
source = e.source
if source is not None:
source = str(source)
else:
source = ''
SingleServerIRCBot._dispatcher(self, c, e)
def restart_program(self):
print ('Restarting')
subprocess.Popen("/home/pi/pajabot/bot/pajabot.py", shell=False)
SingleServerIRCBot.die(self, 'By your command')
exit("updating")