From e050803d4b36bf79fbf55ac7eaaf45c90054eeba Mon Sep 17 00:00:00 2001 From: Gordon Shumway <39967334+oddluck@users.noreply.github.com> Date: Tue, 12 Mar 2019 01:33:56 -0400 Subject: [PATCH] use fake_useragent --- WikiLeaf/plugin.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/WikiLeaf/plugin.py b/WikiLeaf/plugin.py index 4d22b5c..5ef74b1 100644 --- a/WikiLeaf/plugin.py +++ b/WikiLeaf/plugin.py @@ -14,6 +14,7 @@ import supybot.ircmsgs as ircmsgs import requests import re from bs4 import BeautifulSoup +from fake_useragent import UserAgent try: from supybot.i18n import PluginInternationalization @@ -31,13 +32,14 @@ class WikiLeaf(callbacks.Plugin): """ Returns strain information from WikiLeaf. Searches powered by DuckDuckGo. """ - strain = strain.replace(" ", "-").replace("#", "").lower() searchurl = "https://duckduckgo.com/html/?q={0} site: wikileaf.com/strain".format(strain) + ua = UserAgent() + header = {'User-Agent':str(ua.random)} try: - search = requests.get(searchurl) + search = requests.get(searchurl, headers=header) soup = BeautifulSoup(search.text) url = re.sub('\s+', '', soup.find("a", class_="result__url").getText()) - data = requests.get("https://{0}".format(url)) + data = requests.get("https://{0}".format(url), headers=header) soup = BeautifulSoup(data.text) name = re.sub('\s+', ' ', soup.find("h1", itemprop="name").getText()) straininfo = re.sub('\s+', ' ', soup.find("div", class_="product-info-line cannabis").getText()) @@ -50,4 +52,3 @@ class WikiLeaf(callbacks.Plugin): strain = wrap(strain, ['text']) Class = WikiLeaf -