From c7ace886dede07cab8034337f136479ef32051bc Mon Sep 17 00:00:00 2001
From: Thuban <thuban@yeuxdelibad.net>
Date: Mon, 21 Aug 2017 13:40:38 +0000
Subject: [PATCH] support logrotate

---
 vilain.py |   14 +++++++++-----
 1 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/vilain.py b/vilain.py
index 6294620..d36c59f 100755
--- a/vilain.py
+++ b/vilain.py
@@ -24,6 +24,7 @@
 import configparser
 import re
 import logging
+import logging.handlers
 import subprocess
 import asyncio
 import time
@@ -38,10 +39,13 @@
     sys.exit(1)
 
 # Configure logging
+log_handler = logging.handlers.WatchedFileHandler(LOGFILE)
+formatter = logging.Formatter(
+        '%(asctime)s %(module)s:%(funcName)s:%(message)s',
+        '%b %d %H:%M:%S')
+log_handler.setFormatter(formatter)
 logger = logging.getLogger(__name__)
-logging.basicConfig(filename=LOGFILE,
-                    format='%(asctime)s %(module)s:%(funcName)s:%(message)s',
-                    datefmt='%H:%M:%S')
+logger.addHandler(log_handler)
 logger.setLevel(logging.INFO)
 
 # functions
@@ -164,7 +168,7 @@
         record time when this IP has been seen in ip_seen_at = { ip:{'time':<time>,'count':<counter} }
         and ban with pf
         """
-        logger.info('ban_ips sarted')
+        logger.info('ban_ips started')
         while True:
             ip_item = await self.bad_ip_queue.get()
             logger.debug('ban_ips awake')
@@ -185,7 +189,7 @@
         """
         check old ip in ip_seen_at : remove older than watch_while
         """
-        logger.info('clean_ips sarted with sleeptime={}'.format(self.sleeptime))
+        logger.info('clean_ips started with sleeptime={}'.format(self.sleeptime))
         while True:
             await asyncio.sleep(self.watch_while)
             to_remove = []

--
Gitblit v1.9.3