Skip to content

Commit

Permalink
fixed youtube results
Browse files Browse the repository at this point in the history
  • Loading branch information
artyshko committed Feb 8, 2019
1 parent bb159b2 commit ded66af
Showing 1 changed file with 13 additions and 1 deletion.
14 changes: 13 additions & 1 deletion youtube.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,10 @@ def __init__(self):
self.__query = ''
self.__host = 'https://www.youtube.com/'
self.__url = self.__host + 'results?search_query='
self.headers = {
'User-Agent':
'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/39.0.2171.95 Safari/537.36'
}
self.__result = []


Expand All @@ -41,6 +45,13 @@ def getResult(self,i=0):
def getFullResult(self):
return self.__result

def removeInvallidLinks(self):
temp = []
for item in self.getFullResult():
if 40 < len(item) < 50:
temp.append(item)
self.__result = temp


def get(self,text):
'''
Expand All @@ -52,13 +63,14 @@ def get(self,text):
logging.info(f"Finding")

request = self.__url + str(text).replace(' ','+')
response = requests.get(request)
response = requests.get(request, headers=self.headers)
soup = BeautifulSoup(response.text,'lxml')
self.__result = []

for link in soup.findAll(attrs={'class': 'yt-uix-tile-link'}):
self.__result.append(self.__host + link['href'])

self.removeInvallidLinks()
return self.__result


Expand Down

0 comments on commit ded66af

Please sign in to comment.