added youtube sentiment

Author: mjhea0

31_youtube_sentiment.py

@@ -0,0 +1,77 @@
+importsys
+importrequests
+frombs4importBeautifulSoupasbs4
+
+"""
+Example usage:
+
+$ python 31_youtube_sentiment.py https://www.youtube.com/watch?v=_vrAjAHhUsA
+"""
+
+
+defget_arguments():
+iflen(sys.argv) is2:
+returnsys.argv[1]
+else:
+print('Specify at least 1 argument')
+sys.exit()
+
+
+defget_comments(url):
+html=requests.get('https://plus.googleapis.com/u/0/_/widget/render/comments?first_party_property=YOUTUBE&href='+url)
+soup=bs4(html.text, 'html.parser')
+return [comment.stringforcommentinsoup.findAll('div', class_='Ct')]
+
+
+defcalculate_sentiment(comments):
+positive=0
+negative=0
+negative_words= [
+'hate', 'hated', 'dislike', 'disliked', 'awful', 'terrible', 'bad',
+'painful', 'worst', 'suck', 'rubbish', 'sad', 'sodding'
+ ]
+positive_words= [
+'love', 'loved', 'like', 'liked', 'awesome', 'amazing', 'good',
+'great', 'excellent', 'brilliant', 'cool'
+ ]
+forcommentincomments:
+ifcommentisNone:
+continue
+else:
+forwordincomment.split(' '):
+ifwordinnegative_words:
+negative+=1
+ifwordinpositive_words:
+positive+=1
+return{'positive': positive, 'negative': negative}
+
+
+defmain():
+url=get_arguments()
+ifurl:
+comments=get_comments(url)
+iflen(comments) <=0:
+print('This video has no comments.')
+sys.exit()
+sentiment=calculate_sentiment(comments)
+positive_score=sentiment['positive']
+negative_score=sentiment['negative']
+total_score=positive_score+negative_score
+ifpositive_score>negative_score:
+print('This video is generally positive:')
+print('{0} positive /{1} total hits'.format(
+positive_score, total_score))
+elifnegative_score>positive_score:
+print('This video is generally negative:')
+print ('{0} negative /{1} total hits'.format(
+negative_score, total_score))
+else:
+print('This video is mutual:')
+print('{0} positive{1} negative'.format(
+positive_score, negative_score))
+else:
+print('No url supplied')
+
+
+if__name__=='__main__':
+main()

readme.md

@@ -30,3 +30,4 @@
 1.**28_income_tax_calculator.py**: Income tax calculator via [Taxee](http://taxee.io/)
 1.**29_json_to_yaml.py**: Convert JSON to YAML
 1.**30_fullcontact.py**: Call the [FullcContact](https://www.fullcontact.com/developer/) API
+1.**31_youtube_sentiment.py**: Calculate sentiment score from the comments of a Youtube video

requirements.txt

@@ -1,3 +1,4 @@
+beautifulsoup4==4.4.1
 PyYAML==3.11
 requests==2.7.0
 wheel==0.24.0