import datetime, glob, os, shutil, time
from src import ModuleManager, utils

BACKUP_INTERVAL = 60*60 # 1 hour
BACKUP_COUNT = 5

class Module(ModuleManager.BaseModule):
    def on_load(self):
        now = datetime.datetime.now()
        until_next_hour = 60-now.second
        until_next_hour += ((60-(now.minute+1))*60)

        self.timers.add("database-backup", BACKUP_INTERVAL,
            time.time()+until_next_hour)

    @utils.hook("timer.database-backup")
    def backup(self, event):
        location =  self.bot.database.location
        files = glob.glob("%s.*" % location)
        files = sorted(files)

        if len(files) == 5:
            os.remove(files[0])

        suffix = datetime.datetime.now().strftime("%y-%m-%d.%H:%M:%S")
        backup_file = "%s.%s" % (location, suffix)
        shutil.copy2(location, backup_file)

        event["timer"].redo()