From aeaa5ac00a5436a3a7e6f7ae59f807adaa5ab45f Mon Sep 17 00:00:00 2001
From: Ben Harris <ben@tilde.team>
Date: Fri, 26 Aug 2022 13:51:53 -0400
Subject: [PATCH] tidy up clear_weekly_users.py

---
 clear_weekly_users.py | 15 +++++----------
 data_manager.py       |  4 ++++
 2 files changed, 9 insertions(+), 10 deletions(-)

diff --git a/clear_weekly_users.py b/clear_weekly_users.py
index 2e4308b..cea18ef 100644
--- a/clear_weekly_users.py
+++ b/clear_weekly_users.py
@@ -1,11 +1,6 @@
-import os
-import sqlite3
+#!/usr/bin/env python3
 
-game_dir = os.path.dirname(os.path.realpath(__file__))
-garden_db_path = os.path.join(game_dir, 'sqlite/garden_db.sqlite')
-conn = sqlite3.connect(garden_db_path)
-c = conn.cursor()
-c.execute("DELETE FROM visitors")
-print("Cleared weekly users")
-conn.commit()
-conn.close()
+from data_manager import DataManager
+
+d = DataManager()
+d.clear_weekly_visitors()
diff --git a/data_manager.py b/data_manager.py
index 84ee7b6..487ea17 100644
--- a/data_manager.py
+++ b/data_manager.py
@@ -255,3 +255,7 @@ class DataManager(object):
         else:
             visitor_block = 'nobody :('
         return visitor_block
+
+    def clear_weekly_visitors(self):
+        self.cursor.execute("DELETE FROM visitors")
+        self.conn.commit()