class CustomSearchApi(): API_KEY = ConfigLoader.load("google_api_key") SEARCH_ENGINE_ID = ConfigLoader.load("search_engine_id") URL = "https://www.googleapis.com/customsearch/v1" class NetworkError(RuntimeError): def __init__(self, url, status_code, message=""): self.url = url self.status_code = status_code self.message = message if message else \ "%d while requesting %s" % (status_code, url) def list(self, q, **kwargs): Log.d("Searching: %s" % q) if not q: return {} args = dict(kwargs) args.update({ "key": self.API_KEY, "cx": self.SEARCH_ENGINE_ID, "safe": "medium", "q": q, }) with http_request("GET", self.URL, params=args) as response: if response.status_code != 200: Log.e("Failed while list: %d" % response.status_code) if response.text: raise self.NetworkError(url=self.URL, status_code=response.status_code, message=response.text) else: raise self.NetworkError(url=self.URL, status_code=response.status_code) else: return response.json()
def test_pass_bad_project_arg_to_config_loader(): '''config_loader should raise an exception when a bad cli arg is passed''' arg_type = "projects" arg_val = "BADARGUMENT" config = ConfigLoader() with pytest.raises(KeyError) as ke: config.validate(arg_type, arg_val) assert KeyError in str(ke.value)
def run(self): url = ConfigLoader.load("paw_app")["url"] secret = ConfigLoader.load("paw_app")["webhook_secret"] app = Flask(__name__) def _webhook_view(): return self._on_webhook() app.add_url_rule("/%s" % secret, view_func = _webhook_view, methods = ["POST"]) global flask_app flask_app = app self._bot.setWebhook("%s/%s" % (url, secret), max_connections = 1)
class StandaloneApp(): TELEGRAM_TOKEN = ConfigLoader.load("telegram_bot_token") def __init__(self): Log.i("Initializing standalone app") self._bot = telepot.Bot(self.TELEGRAM_TOKEN) self._answerer = telepot.helper.Answerer(self._bot) def run(self): import time self._start() Log.i("Running...") while True: time.sleep(10) def _start(self): Log.i("Starting app") def _listener(msg): self._on_message(msg) def _inline_listener(msg): self._on_inline_message(msg) self._bot.setWebhook("") self._bot.message_loop({ "chat": _listener, "inline_query": _inline_listener, }) def _on_message(self, msg): MessageHandler(self._bot, msg).handle() def _on_inline_message(self, msg): InlineMessageHandler(self._bot, self._answerer, msg).handle()
def _ensure_allowed_users(user): allows = ConfigLoader.load("allow_only_users") if not allows: # Empty list == allow all return # Allow if either id or username is whitelisted if user["id"] in allows: return elif "username" in user and user["username"] in allows: return else: raise _WhitelistException()
class PawApp(): TELEGRAM_TOKEN = ConfigLoader.load("telegram_bot_token") def __init__(self): Log.i("Initializing PAW app") self._init_paw_telepot() self._bot = telepot.Bot(self.TELEGRAM_TOKEN) def run(self): url = ConfigLoader.load("paw_app")["url"] secret = ConfigLoader.load("paw_app")["webhook_secret"] app = Flask(__name__) def _webhook_view(): return self._on_webhook() app.add_url_rule("/%s" % secret, view_func = _webhook_view, methods = ["POST"]) global flask_app flask_app = app self._bot.setWebhook("%s/%s" % (url, secret), max_connections = 1) def _init_paw_telepot(self): # You can leave this bit out if you're using a paid PythonAnywhere # account proxy_url = "http://proxy.server:3128" telepot.api._pools = { "default": urllib3.ProxyManager(proxy_url = proxy_url, num_pools = 3, maxsize = 10, retries = False, timeout = 30), } telepot.api._onetime_pool_spec = (urllib3.ProxyManager, dict(proxy_url = proxy_url, num_pools = 1, maxsize = 1, retries = False, timeout = 30)) # end of the stuff that's only needed for free accounts def _on_webhook(self): update = request.get_json() if "message" in update: # message request MessageHandler(self._bot, update["message"]).handle() elif "inline_query" in update: # inline request InlineMessageNoThreadHandler(self._bot, update["inline_query"]) \ .handle() return "OK"
class StandaloneApp: TELEGRAM_TOKEN = ConfigLoader.load("telegram_bot_token") def __init__(self): Log.i("Initializing standalone app") self._bot = telepot.Bot(self.TELEGRAM_TOKEN) def run(self): import time self._start() Log.i("Running...") while True: time.sleep(10) def _start(self): Log.i("Starting app") def _listener(msg): self._on_message(msg) def _callback_query_listener(msg): self._on_callback_query(msg) self._bot.setWebhook("") self._bot.message_loop({ "chat": _listener, "callback_query": _callback_query_listener, }) def _on_message(self, msg): MessageHandler(self._bot, msg, self._make_session_class()).handle() def _on_callback_query(self, msg): CallbackQueryHandler(self._bot, msg, self._make_session_class()).handle() def _make_session_class(self): sqlite_engine = create_engine("sqlite:///poll.db", echo=True) return sessionmaker(bind=sqlite_engine)
def test_pass_good_project_arg_to_config_loader(): '''config_loader should return 'true' when a good cli arg and active project is passed''' arg_type = "platforms" arg_val = "usa" config = ConfigLoader() assert config.validate(arg_type, arg_val) == 'usa'
def test_pass_good_project_arg_to_config_loader(): '''config_loader should return 'true' when a good cli arg and active project is passed''' arg_type = "environments" arg_val = "dev" config = ConfigLoader() assert config.validate(arg_type, arg_val) == 'dev'
def test_instantiate_config_loader(): '''init should read in the config.ini file successfully''' config = ConfigLoader() assert isinstance(config, ConfigLoader) == True
def test_pass_good_project_arg_to_config_loader(): '''config_loader should return 'false' when a good cli arg inactive project is passed''' arg_type = "projects" arg_val = "media-core-ui" config = ConfigLoader() assert config.validate(arg_type, arg_val) == 'false'
class PawApp(): TELEGRAM_TOKEN = ConfigLoader.load("telegram_bot_token") def __init__(self): Log.i("Initializing PAW app") self._init_paw_telepot() self._bot = telepot.Bot(self.TELEGRAM_TOKEN) def run(self): url = ConfigLoader.load("paw_app")["url"] secret = ConfigLoader.load("paw_app")["webhook_secret"] app = Flask(__name__) def _webhook_view(): return self._on_webhook() app.add_url_rule("/%s" % secret, view_func=_webhook_view, methods=["POST"]) global flask_app flask_app = app self._bot.setWebhook("%s/%s" % (url, secret), max_connections=1) def _init_paw_telepot(self): # You can leave this bit out if you're using a paid PythonAnywhere # account proxy_url = "http://proxy.server:3128" telepot.api._pools = { "default": urllib3.ProxyManager(proxy_url=proxy_url, num_pools=3, maxsize=10, retries=False, timeout=30), } telepot.api._onetime_pool_spec = (urllib3.ProxyManager, dict(proxy_url=proxy_url, num_pools=1, maxsize=1, retries=False, timeout=30)) # end of the stuff that's only needed for free accounts def _on_webhook(self): update = request.get_json() self._handle_update(update) return "OK" def _handle_update(self, update): if "update_id" in update: if not self._should_process_update(update["update_id"]): return if "message" in update: # message request MessageHandler(self._bot, update["message"], self._make_session_class()).handle() elif "callback_query" in update: # inline request CallbackQueryHandler(self._bot, update["callback_query"], self._make_session_class()).handle() def _should_process_update(self, update_id): import datetime now = datetime.datetime.utcnow() dt = datetime.timedelta(weeks=1) from_time = now - dt with model.open_session(self._make_session_class()) as s: count = s.query(model.HandledUpdate) \ .filter(model.HandledUpdate.update_id == update_id) \ .filter(model.HandledUpdate.created_at >= from_time) \ .count() if count == 0: # Add this update m = model.HandledUpdate(update_id=update_id) s.add(m) # Cleanup old ones s.query(model.HandledUpdate) \ .filter(model.HandledUpdate.created_at < from_time) \ .delete(synchronize_session = False) return (count == 0) def _make_session_class(self): sqlite_engine = create_engine("sqlite:///poll.db", echo=True) return sessionmaker(bind=sqlite_engine)