Merge branch 'master' of schneider/schneiderbot-tg into master

.gitignore

@@ -1,4 +1,5 @@
 token
+deaths.db
 
 # ---> Python
 # Byte-compiled / optimized / DLL files

db.py

@@ -0,0 +1,34 @@
+"""DB interface"""
+import sqlite3
+
+class Db():
+    """ Abstract interface for db actions """
+    def __init__(self):
+        self.conn = sqlite3.connect("deaths.db")
+        table_exist = self.conn.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='death'")
+        if table_exist.fetchall() == []:
+            self.conn.execute("CREATE TABLE death(name)")
+
+    def __del__(self):
+        self.conn.commit()
+
+    def is_dead(self, user):
+        """Reports whether a user is already dead"""
+        print(str(user))
+        res = self.conn.execute("select * from death where name = ?", [user])
+        return res.fetchall() != []
+
+    def kill(self, user):
+        """Kills a user"""
+        self.conn.execute("insert into death(name) values(?)", [user])
+
+    def revive(self, user):
+        """Revives a user"""
+        self.conn.execute("delete from death where name = ?", [user])
+
+    def get_dead_bodies(self):
+        """List the dead"""
+        names = []
+        for name in self.conn.execute("select * from death").fetchall():
+            names.append(name[0])
+        return ", ".join(names)

schneiderbot.py

@@ -3,6 +3,7 @@
 
 from telegram.ext import Updater, CommandHandler, MessageHandler, Filters
 from telegram import ParseMode
+from db import Db
 import logging
 import os
 import random
@@ -128,23 +129,44 @@ def sendDog(bot, update):
 
 def kill(bot, update, sudocall = False):
     """kill me pls"""
+    db = Db()
+    user = update.message.from_user.first_name
+    if db.is_dead(user):
+        if sudocall:
+           update.message.reply_text("%s killed again" % user)
+           return
+        update.message.reply_text("%s is already dead" % user)
+        return
+
     message = update.message.from_user.first_name + " died"
     if sudocall:
         message += " for real"
+    db.kill(user)
     update.message.reply_text(message)
 
 def revive(bot, update, sudocall = False):
     """unkill me pls"""
-    message = update.message.from_user.first_name + " is living again"
+    db = Db()
+    user = update.message.from_user.first_name
+    if not db.is_dead(user):
+        update.message.reply_text("Maybe %s should have a litte 'accident' before" % user)
+        return
     if sudocall:
-        message += " for real"
-    update.message.reply_text(message)
+        update.message.reply_text("%s is living again, for real" % user)
+        db.revive(user)
+        return
+    update.message.reply_text("You fool! You cannot revive a dead person!")
 
 SUDOCMDS = {
     "kill" : kill,
     "revive" : revive
 }
 
+def graveyard(bot, update):
+    """List the dead"""
+    db = Db()
+    update.message.reply_text("Here are the dead people:\n%s" % db.get_dead_bodies())
+
 def sudo(bot, update, args):
     """for real"""
     if not args :
@@ -267,6 +289,7 @@ def main():
     dp.add_handler(CommandHandler("kill", kill))
     dp.add_handler(CommandHandler("revive", revive))
     dp.add_handler(CommandHandler("simon", simon))
+    dp.add_handler(CommandHandler("graveyard", graveyard))
 
 
     # log all errors
@@ -283,4 +306,3 @@ def main():
 
 if __name__ == '__main__':
     main()
-