Merge 849e48b into bb51e3e

pastepwn/core/pastepwn.py

@@ -28,6 +28,7 @@ def __init__(self, database=None, proxies=None, store_all_pastes=True):
         self.paste_queue = Queue()
         self.action_queue = Queue()
         self.error_handlers = list()
+        self.onstart_handlers = list()
         self.__exception_event = Event()
         self.__request = Request(proxies)  # initialize singleton
 
@@ -89,6 +90,13 @@ def add_error_handler(self, error_handler):
 
         self.error_handlers.append(error_handler)
 
+    def add_onstart_handler(self, onstart_handler):
+        if not callable(onstart_handler):
+            self.logger.error("The onstart handler you passed is not a function!")
+            return
+
+        self.onstart_handlers.append(onstart_handler)
+
     def start(self):
         """Starts the pastepwn instance"""
         if self.__exception_event.is_set():
@@ -102,6 +110,12 @@ def start(self):
         self.paste_dispatcher.start()
         self.action_handler.start()
 
+        for onstart_handler in self.onstart_handlers:
+            try:
+                onstart_handler()
+            except Exception as e:
+                self.logger.error("Onstart hander %s failed with error: %s. Pastepwn is still running." % (onstart_handler.__name__, e))
+
     def stop(self):
         """Stops the pastepwn instance"""
         self.scraping_handler.stop()