onjen/discord_checkin_bot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add checking and autocheckout

Browse Source
This commit is contained in:
Johannes Rothe 2021-12-29 00:56:42 +01:00
parent fb4318693b
commit aa586fcd4a
5 changed files with 125 additions and 82 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

87
checkin_bot.py Normal file
View File

@ -0,0 +1,87 @@
import sqlite3
from datetime import datetime, timedelta
from typing import List, Tuple
AUTO_CHECKOUT_HOURS = 6
class CheckinDB:
def __init__(self, db_name: str) -> None:
self.con = sqlite3.connect(db_name)
self.con.row_factory = sqlite3.Row
self.try_create_tables()
def try_create_tables(self) -> bool:
try:
with self.con:
self.con.execute(
"CREATE TABLE checkin (ID INTEGER PRIMARY KEY AUTOINCREMENT, NAME VARCHAR UNIQUE, IS_PRESENT BOOL, UPDATED INT)"
)
print("Creating new table")
except sqlite3.OperationalError:
print("Table already exists in database, nothing to create.")
return True
def update(self, name: str, is_present: bool) -> bool:
present_int = 1 if is_present else 0
with self.con:
self.con.execute(
f"INSERT INTO checkin (NAME, IS_PRESENT, UPDATED) VALUES ('{name}', '{present_int}', strftime('%s','now')) ON CONFLICT(name) DO UPDATE SET is_present=excluded.is_present"
)
return True
return False
def get_full_entries(self) -> List[Tuple]:
with self.con:
return [
(x["name"], x["is_present"], x["updated"])
for x in self.con.execute(
"SELECT name, is_present, updated FROM checkin ORDER BY name"
)
]
def get_checkin_names(self) -> List[str]:
with self.con:
return [
x["name"]
for x in self.con.execute(
"SELECT name FROM checkin WHERE is_present ORDER BY name"
)
]
def get_checkin_count(self) -> int:
with self.con:
return self.con.execute(
"SELECT COUNT(name) FROM checkin WHERE is_present"
).fetchone()[0]
def clear_rows(self) -> None:
with self.con:
self.con.execute("DELETE FROM checkin")
def get_updated_for_name(self, name: str) -> datetime:
with self.con:
return datetime.fromtimestamp(
float(
self.con.execute(
f"SELECT updated FROM checkin WHERE name = '{name}'"
).fetchone()[0]
)
)
def auto_checkout(self) -> None:
filter_time = (
datetime.now() - timedelta(hours=AUTO_CHECKOUT_HOURS)
).timestamp()
with self.con:
self.con.execute(
f"UPDATE checkin SET IS_PRESENT = 0 WHERE updated <= {filter_time}"
)
def main() -> None:
counter_bot = CheckinDB("checkin.db")
if __name__ == "__main__":
main()

65
counter_bot.py
View File

@ -1,65 +0,0 @@
import sqlite3
from typing import List, Tuple
class CounterBot:
def __init__(self, db_name: str) -> None:
self.con = sqlite3.connect(db_name)
self.con.row_factory = sqlite3.Row
self.try_create_tables()
def try_create_tables(self) -> bool:
try:
with self.con:
self.con.execute(
"CREATE TABLE presence (ID INTEGER PRIMARY KEY AUTOINCREMENT, NAME VARCHAR UNIQUE, IS_PRESENT BOOL)"
)
print("Creating new table")
except sqlite3.OperationalError:
print("Table already exists in database, nothing to create.")
return True
def update_presence(self, name: str, is_present: bool) -> bool:
present_int = 1 if is_present else 0
with self.con:
self.con.execute(
f"INSERT INTO presence (NAME, IS_PRESENT) VALUES ('{name}', '{present_int}') ON CONFLICT(name) DO UPDATE SET is_present=excluded.is_present"
)
return True
return False
def get_full_entries(self) -> List[Tuple]:
with self.con:
return [
(x["name"], x["is_present"])
for x in self.con.execute(
"SELECT name, is_present FROM presence ORDER BY name"
)
]
def get_present_names(self) -> List[str]:
with self.con:
return [
x["name"]
for x in self.con.execute(
"SELECT name FROM presence WHERE is_present ORDER BY name"
)
]
def get_present_count(self) -> int:
with self.con:
return self.con.execute(
"SELECT COUNT(name) FROM presence WHERE is_present"
).fetchone()[0]
def clear_rows(self) -> None:
with self.con:
self.con.execute("DELETE FROM presence")
def main() -> None:
counter_bot = CounterBot("counter.db")
if __name__ == "__main__":
main()

2
requirements.txt Normal file
View File

@ -0,0 +1,2 @@
pytest
discord

36
test_checkin_bot.py Normal file
View File

@ -0,0 +1,36 @@
from datetime import timedelta, datetime
from .checkin_bot import CheckinDB
TEST_DB_NAME = "test.db"
checkin_db = CheckinDB(TEST_DB_NAME)
checkin_db.clear_rows()
def test_get_present_count() -> None:
checkin_db.update("Alice", True)
checkin_db.update("Bob", True)
checkin_db.update("Bob", False)
checkin_db.update("Bob", True)
assert checkin_db.get_checkin_count() == 2
def test_present_names() -> None:
assert checkin_db.get_checkin_names() == ["Alice", "Bob"]
def test_updated_timestamp() -> None:
assert checkin_db.get_updated_for_name(
"Alice"
) >= datetime.now() - timedelta(minutes=1)
def test_auto_checkout() -> None:
assert "Alice" in checkin_db.get_checkin_names()
old_timestamp = (datetime.now() - timedelta(hours=6)).timestamp()
with checkin_db.con:
checkin_db.con.execute(
f"UPDATE checkin SET updated = {old_timestamp} WHERE name = 'Alice'"
)
checkin_db.auto_checkout()
assert "Alice" not in checkin_db.get_checkin_names()

17
test_counter_bot.py
View File

@ -1,17 +0,0 @@
import sqlite3
from .counter_bot import CounterBot
TEST_DB_NAME = "test.db"
counter_bot = CounterBot(TEST_DB_NAME)
counter_bot.clear_rows()
def test_get_present_count() -> None:
counter_bot.update_presence("Alice", True)
counter_bot.update_presence("Bob", True)
counter_bot.update_presence("Bob", False)
counter_bot.update_presence("Bob", True)
assert counter_bot.get_present_count() == 2
def test_present_names() -> None:
assert counter_bot.get_present_names() == ["Alice", "Bob"]