Skip to content

Commit 8ff765c

Browse files
committed
save privileged list periodically
1 parent 18dea30 commit 8ff765c

3 files changed

Lines changed: 31 additions & 4 deletions

File tree

Source/Bot.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def __init__(self, bot_name, client, commands, room_ids, background_tasks=[], bo
3030
self.github_link = github_link
3131
self.background_task_manager = BackgroundTaskManager(background_tasks)
3232
self.chatcommunicate = Chatcommunicate(bot_name, CommandManager(commands, self.rooms))
33-
self.save_directory = os.path.expanduser("~") + "." + self.name.lower()
33+
self.save_directory = os.path.expanduser("~") + "/" + "." + self.name.lower() + "/"
3434

3535
def add_background_task(self, background_task, interval=30, restart=True):
3636
self.background_task_manager.add_background_task(background_task)
@@ -41,6 +41,9 @@ def add_essential_background_tasks(self, restart=True):
4141
self.add_background_task(BackgroundTask(self.chatcommunicate.command_manager.cleanup_finished_commands, interval=3))
4242
self.add_background_task(BackgroundTask(self.shutdown_check, interval=5))
4343

44+
for each_room in self.rooms:
45+
self.add_background_task(BackgroundTask(each_room.save_privileged_users))
46+
4447
self.background_task_manager.restart_tasks()
4548

4649
def join_rooms(self, watch_callback):
@@ -65,8 +68,13 @@ def add_privilege_type(self, privilege_level, privilege_name):
6568
for each_room in self.rooms:
6669
each_room.add_privilege_type(privilege_level, privilege_name)
6770

71+
def load_privileged_user_list(self):
72+
for each_room in self.rooms:
73+
each_room.load_privileged_users()
74+
6875
def start_bot(self):
6976
self.join_rooms(self.chatcommunicate.handle_message)
77+
self.load_privileged_user_list()
7078
self.watch_rooms()
7179
self.background_task_manager.start_tasks()
7280

Source/CommandPrivilegeUser.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
#
2+
# CommandPrivilegeUser.py
3+
# Botpy
4+
#
5+
# Created by Ashish Ahuja on 17th September 2017.
6+
#
7+
#
8+
9+
from Command import *
10+
11+
class CommandPrivilegeUser(Command):
12+
def usage():
13+
return ["privilege * *", "addpriv * *"]
14+
15+
def privileges(self):
16+
return 1
17+
18+
def run(self):
19+

Source/test.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,10 @@
1111

1212
bot = Bot.Bot("sock", client, commands, [123602])
1313

14-
bot.add_essential_background_tasks()
15-
1614
bot.start_bot()
1715

16+
bot.add_essential_background_tasks()
17+
1818
bot.add_privilege_type(1, "owner")
1919

20-
bot.rooms[0].add_privileged_user(4688119, bot.rooms[0].get_privilege_type_by_name("owner"))
20+
#bot.rooms[0].add_privileged_user(4688119, bot.rooms[0].get_privilege_type_by_name("owner"))

0 commit comments

Comments
 (0)