def test_clean_links_one_link(self):
text = "Ananas banana kołysał https://youtu.be/6kS65TbKjeU?t=49 fajne"
result = "Ananas banana kołysał fajne"
self.assertEqual(Tools.clean_links(text), result)
def test_clean_links_two_links(self):
text = "Ananas https://t.co/5mCBmQ52w8 banana kołysał https://youtu.be/_uJQZ8QmoKo fajne"
result = "Ananas banana kołysał fajne"
self.assertEqual(Tools.clean_links(text), result)
def test_clean_links_end_link(self):
text = "Ananas banana kołysał fajne https://t.co/jlzzctiGeC"
result = "Ananas banana kołysał fajne"
self.assertEqual(Tools.clean_links(text), result)
def report():
if not request.args.get("tweets") and not request.args.get(
"retweets") and not request.args.get("replies"):
return redirect(
url_for(
'index',
error=
"Error: At least one \"include\" option need to be checked."))
nickname = request.args.get('nickname')
try:
user = api.GetUser(screen_name=nickname)
except TwitterError:
return redirect(url_for('index', error="Error: User not found."))
timeline = api.GetUserTimeline(
screen_name=nickname,
count=200,
trim_user=True,
include_rts=request.args.get("retweets"),
exclude_replies=not request.args.get("replies"))
if not request.args.get("tweets"):
timeline = Tools.cut_tweets(timeline)
if request.args.get("links"):
old_timeline = timeline.copy()
i = 0
for tweet in old_timeline:
timeline[i].full_text = Tools.clean_links(tweet.full_text)
if timeline[i].full_text == "":
del timeline[i]
else:
i += 1
if len(timeline) < 3:
return redirect(
url_for('index',
error="Error: Too few tweets with given criteria found."))
replies_script, replies_div = components(create_replies_graph(timeline))
favorites_script, favorites_div = components(
create_favorites_graph(timeline, 10))
posts_in_days_script, posts_in_days_div = components(
create_posts_in_days_graph(timeline))
length_script, length_div = components(create_length_graph(timeline, 10))
posts_in_hours_script, posts_in_hours_div = components(
create_posts_in_hours_graph(timeline))
tfidf_script, tfidf_div = components(create_tfidf_graph(timeline))
wordcloud(timeline, user.id_str)
# grab the static resources
js_resources = INLINE.render_js()
css_resources = INLINE.render_css()
avatar_url = "https://twitter.com/" + nickname + "/profile_image?size=bigger"
r = requests.get(avatar_url)
avatar_path = "static/temp/avatar" + user.id_str + ".png"
with open(avatar_path, 'wb') as f:
f.write(r.content)
try:
following_tooltip = len(api.GetFriendIDs(user.id))
following = Tools.human_format(following_tooltip)
except TwitterError:
following = "???"
following_tooltip = "Reached api limit"
tweets = user.statuses_count
followers = user.followers_count
likes = user.favourites_count
posts = len(timeline)
date_last = Tools.human_format_date(timeline[0].created_at)
date_first = Tools.human_format_date(timeline[len(timeline) -
1].created_at)
# render template
html = render_template(
"report.html",
nickname=nickname,
tweets=Tools.human_format(tweets),
following=following,
followers=Tools.human_format(followers),
likes=Tools.human_format(likes),
tweets_tooltip=tweets,
following_tooltip=following_tooltip,
followers_tooltip=followers,
likes_tooltip=likes,
replies_script=replies_script,
replies_div=replies_div,
favorites_script=favorites_script,
favorites_div=favorites_div,
posts_in_days_script=posts_in_days_script,
posts_in_days_div=posts_in_days_div,
posts_in_hours_script=posts_in_hours_script,
posts_in_hours_div=posts_in_hours_div,
js_resources=js_resources,
css_resources=css_resources,
length_script=length_script,
length_div=length_div,
tfidf_script=tfidf_script,
tfidf_div=tfidf_div,
pid="temp/WC" + user.id_str + ".png",
avatar="temp/avatar" + user.id_str + ".png",
tweets_checked=request.args.get("tweets"),
replies_checked=request.args.get("replies"),
retweets_checked=request.args.get("retweets"),
links_checked=request.args.get("links"),
posts=Tools.human_format(posts),
date_interval=date_first + " - " + date_last,
)
return encode_utf8(html)
