From 38ba4647a04ebf482c993c9b46a04eb7944e6cca Mon Sep 17 00:00:00 2001
From: flifloo <flifloo@gmail.com>
Date: Thu, 5 Nov 2020 14:37:37 +0100
Subject: [PATCH] Add check for events

---
 administrator/utils.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/administrator/utils.py b/administrator/utils.py
index 6908e4e..fba2bdf 100644
--- a/administrator/utils.py
+++ b/administrator/utils.py
@@ -2,6 +2,9 @@ import re
 from datetime import timedelta
 
 from discord.ext.commands import BadArgument
+from sqlalchemy.orm import Session
+
+import db
 
 
 def time_pars(s: str) -> timedelta:
@@ -23,3 +26,19 @@ def seconds_to_time_string(seconds: float) -> str:
     if days > 0 else f"{hours}h {minutes}m {seconds}s"
     if hours > 0 else f"{minutes}m {seconds}s"
     if minutes > 0 else f"{seconds}s"}"""
+
+
+def event_is_enabled(cog: str, guild_id: int, s: Session = None) -> bool:
+    close = False
+    if not s:
+        s = db.Session()
+        close = True
+
+    es = s.query(db.ExtensionState).get((cog, guild_id))
+
+    if close or (es and not es.state):
+        s.close()
+
+    if es:
+        return es.state
+    return True