initial commit

Signed-off-by: call-me-matt <[email protected]>

Author: call-me-matt

Dockerfile

@@ -0,0 +1,10 @@
+FROM python
+
+WORKDIR /usr/src/app
+
+COPY requirements.txt ./
+RUN pip3 install --no-cache-dir -r requirements.txt
+
+#COPY ./src .
+
+CMD [ "python", "./osmChangesetsBot.py" ]

README.md

@@ -0,0 +1,18 @@
+# osmChangeMonitorBot
+
+this is a bot for telegram that monitors changes for a list of openstreetmap users.
+
+you can customize your list and requests stats by sending `/stats`. Additionally, if a user exceeds 300 or 1000 changes in the current month, a notification message is sent.
+
+## installation
+
+ * create a bot token for telegram messenger: https://t.me/botfather
+
+ * add the token to the secrets.env-File
+
+ * use docker-compose to start: `docker-compose up`
+
+## demo
+
+or try it out (only working if my instance is running): https://t.me/osmChangesetBot
+

docker-compose.yml

@@ -0,0 +1,9 @@
+version: '3'
+services:
+  bot:
+    build: .
+    restart: unless-stopped
+    env_file:
+    - secrets.env
+    volumes:
+    - ./src:/usr/src/app

requirements.txt

@@ -0,0 +1,2 @@
+python-telegram-bot
+requests

secrets.env

@@ -0,0 +1 @@
+TELEGRAM_BOT_TOKEN=yourtoken

src/databaseHandler.py

@@ -0,0 +1,132 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+import sqlite3
+import logging
+import datetime
+
+logging.basicConfig(format='[%(levelname)s] %(name)s: %(message)s',level=logging.INFO)
+logger = logging.getLogger("database-handler")
+
+def init():
+    logger.info("initializing database")
+    con = sqlite3.connect('registration.db')
+    db = con.cursor()
+    db.execute("CREATE TABLE IF NOT EXISTS users( \
+                    user TINYTEXT PRIMARY KEY, \
+                    chatid TINYTEXT \
+               )")
+    con.commit()
+    db.execute("CREATE TABLE IF NOT EXISTS osmStats( \
+                user TINYTEXT PRIMARY KEY, \
+                changes TINYTEXT \
+           )")
+    con.commit()
+    db.execute("CREATE TABLE IF NOT EXISTS watchers( \
+                telegramUser TINYTEXT, \
+                osmUser TINYTEXT \
+           )")
+    con.commit()
+    con.close()
+
+def addUser(username, chatid):
+    logger.info("adding telegram user " + username + " to database")
+    con = sqlite3.connect('registration.db')
+    db = con.cursor()
+    db.execute("INSERT OR IGNORE INTO users (user,chatid) VALUES (?,?)",([username,chatid]))
+    con.commit()
+    con.close()
+        
+def removeUser(username):
+    logger.info("removing telegram user " + username + " from database")
+    con = sqlite3.connect('registration.db')
+    db = con.cursor()
+    db.execute("DELETE FROM users WHERE user=?",([username]))
+    db.execute("DELETE FROM watchers WHERE telegramUser=?",([username]))
+    db.execute("DELETE FROM osmStats WHERE user not in (SELECT DISTINCT osmUser from watchers)")
+    con.commit()
+    con.close()
+
+def isUserRegistered(telegramUser):
+    con = sqlite3.connect('registration.db')
+    db = con.cursor()
+    db.execute("SELECT chatid FROM users WHERE user=?",([telegramUser]))
+    entries = db.fetchall()
+    con.close()
+    if entries == None or entries == []:
+        return False
+    return True
+
+def addWatcher(telegramUser, osmUser):
+    logger.info("adding follower " + telegramUser + " for " + osmUser + " to the database")
+    con = sqlite3.connect('registration.db')
+    db = con.cursor()
+    #TODO: only if not existing
+    db.execute("INSERT INTO watchers (telegramUser,osmUser) VALUES (?,?)",([telegramUser,osmUser]))
+    con.commit()
+    con.close()
+
+def getWatcher(osmUser):
+    con = sqlite3.connect('registration.db')
+    db = con.cursor()
+    db.execute("SELECT chatid FROM users WHERE user in (SELECT DISTINCT telegramUser FROM watchers WHERE osmUser=?)",([osmUser]))
+    entries = db.fetchall()
+    con.close()
+    result = []
+    for entry in entries:
+        result.append(entry[0])
+    return result
+
+def removeWatcher(telegramUser, osmUser):
+    logger.info("removing follower " + telegramUser + " for " + osmUser + " from the database")
+    con = sqlite3.connect('registration.db')
+    db = con.cursor()
+    db.execute("DELETE FROM watchers WHERE telegramUser=? AND osmUser=?",([telegramUser,osmUser]))
+    db.execute("DELETE FROM osmStats WHERE user not in (SELECT DISTINCT osmUser from watchers)")
+    con.commit()
+    con.close()
+
+def getOsmUsers(telegramUser=None):
+    con = sqlite3.connect('registration.db')
+    con.row_factory = sqlite3.Row
+    db = con.cursor()
+    if telegramUser:
+        db.execute("SELECT DISTINCT osmUser from watchers WHERE telegramUser=?",([telegramUser]))
+    else:
+        db.execute("SELECT DISTINCT osmUser from watchers")
+    entries = db.fetchall()
+    con.close()
+    result = []
+    for entry in entries:
+        result.append(entry['osmUser'])
+    return result
+    
+def getStats(watcher):
+    logger.debug("getStats for " + str(watcher))
+    con = sqlite3.connect('registration.db')
+    con.row_factory = sqlite3.Row
+    db = con.cursor()
+    db.execute("SELECT user,changes FROM osmStats WHERE user in (SELECT osmUser from watchers WHERE telegramUser=?)",([watcher]))
+    entries = db.fetchall()
+    con.close()
+    result = ""
+    for (user,changes) in entries:
+        result += user + ": " + str(changes) + " \n"
+    return result
+
+def updateStats(user, counter):
+    logger.debug('updating stats for ' + user + ': ' + str(counter))
+    con = sqlite3.connect('registration.db')
+    con.row_factory = sqlite3.Row
+    db = con.cursor()
+    db.execute("SELECT changes FROM osmStats WHERE user=?",([user]))
+    counterOld = db.fetchone()
+    db.execute("INSERT INTO osmStats (user, changes) VALUES(?,?) ON CONFLICT(user) DO UPDATE SET changes=?",([user,counter,counter]))
+    con.commit()
+    con.close()
+    if counterOld == None:
+        return -1
+    else:
+        return counterOld[0]
+
+

src/osmChangesetsBot.py

@@ -0,0 +1,29 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+import logging
+import sys
+import time
+import threading
+
+import telegramHandler
+
+logging.basicConfig(format='[%(levelname)s] %(name)s: %(message)s',level=logging.INFO)
+logger = logging.getLogger("main")
+
+logger.info('starting osmChangesetsBot')
+
+logger.debug('creating telegram-handler thread')
+telegramThread = telegramHandler.telegramHandler()
+telegramThread.daemon = True
+telegramThread.start()
+
+while True:
+    try:
+        time.sleep(5)
+    except KeyboardInterrupt:
+        logger.info('Exiting...')
+        sys.exit()
+    except:
+        raise
+