elitebot.py

@@ -1,15 +1,12 @@
 #!/usr/bin/env python3
 
 import asyncio
-import os
 import sys
 
 from src.bot import Bot
 
 
 def main():
-    os.makedirs('data', exist_ok=True)
-
     if len(sys.argv) < 2:
         print('Usage: python elitebot.py <config_file>')
         sys.exit(1)

src/bot.py

@@ -208,8 +208,9 @@ class Bot:
                     case 'VERSION':
                         await self.ircsend(f'NOTICE {source_nick} :I am a bot version 1.0.0')
                     case '001':
-                        for channel in self.channel_manager.get_channels():
-                            await self.ircsend(f'JOIN {channel[1]}')
+                        await self.ircsend(f'JOIN #YuukiTest')
+                        # for channel in self.channel_manager.get_channels():
+                        #     await self.ircsend(f'JOIN {channel}')
                     case '903':
                         await handle_903(self.ircsend)
                     case _:

src/channel_manager.py

@@ -1,30 +1,49 @@
 #!/usr/bin/env python3
 
-from src.db import Database
-from sqlalchemy import Table, Column, Integer, String, Boolean, MetaData
-
-meta = MetaData()
-channel_table = Table(
-    'Channels',
-    meta,
-    Column('id', Integer, primary_key=True, autoincrement=True),
-    Column('channel', String, unique=True, nullable=False),
-    Column('autojoin', Boolean, default=True),
-)
-db = Database(channel_table, meta)
+import json
+import os
+from os import path
 
 
 class ChannelManager:
     def __init__(self):
-        db.create_table(channel_table.name)
+        self.channels = self._load_channels()
 
-        self.channels = db._load_channels()
+    def _load_channels(self):
+        os.makedirs('data', exist_ok=True)
+        if not path.exists('data/channels.json'):
+            with open('data/channels.json', 'w') as f:
+                json.dump([], f)
+            return []
+        try:
+            with open('data/channels.json', 'r') as f:
+                return json.load(f)
+        except json.JSONDecodeError as e:
+            print(f'Error decoding JSON: {e}')
+            return []
+        except Exception as e:
+            print(f'Error loading channels: {e}')
+            return []
 
     def save_channel(self, channel):
-        db._save_channel(channel)
+        channel = channel.lstrip(':')
+        if channel not in self.channels:
+            self.channels.append(channel)
+            self._write_channels()
 
     def remove_channel(self, channel):
-        db._remove_channel(channel)
+        channel = channel.lstrip(':')
+        if channel in self.channels:
+            self.channels.remove(channel)
+            self._write_channels()
+
+    def _write_channels(self):
+        os.makedirs('data', exist_ok=True)
+        try:
+            with open('data/channels.json', 'w') as f:
+                json.dump(self.channels, f)
+        except Exception as e:
+            print(f'Error saving channels: {e}')
 
     def get_channels(self):
         return self.channels

src/db.py

@@ -1,4 +1,4 @@
-from sqlalchemy import create_engine, Table, MetaData, inspect, update, select, insert, delete
+from sqlalchemy import create_engine, Table, MetaData, inspect, update, select
 from sqlalchemy_utils import database_exists, create_database
 
 
@@ -37,31 +37,3 @@ class Database:
                 return conn.execute(select(self.table).where(self.table.c.name == user)).fetchone()[index]
             else:
                 return -1
-
-    def _load_channels(self):
-        with self.engine.connect() as conn:
-            return conn.execute(select(self.table)).fetchall()
-
-    def _save_channel(self, channel: str):
-        with self.engine.connect() as conn:
-            stmt = select(self.table).where(self.table.c.channel == channel)
-            cnt = len(conn.execute(stmt).fetchall())
-
-            if cnt == 0:
-                conn.execute((
-                    insert(self.table).
-                    values({'channel': channel})
-                ))
-                conn.commit()
-
-    def _remove_channel(self, channel: str):
-        with self.engine.connect() as conn:
-            stmt = select(self.table).where(self.table.c.channel == channel)
-            cnt = len(conn.execute(stmt).fetchall())
-
-            if cnt == 1:
-                conn.execute((
-                    delete(self.table).
-                    where(self.table.c.channel == channel)
-                ))
-                conn.commit()