def test_timesince():
then = datetime(2010, 4, 12, 12, 30, 0)
then_timestamp = 1271075400.0
then_future = datetime(2012, 4, 12, 12, 30, 0)
now = datetime(2010, 5, 15, 1, 50, 0)
now_timestamp = 1273888200.0
# timestamp
assert timesince(then_timestamp, now_timestamp) == "1 month and 2 days"
# basic
assert timesince(then, now) == "1 month and 2 days"
# count
assert timesince(then, now, count=3) == "1 month, 2 days and 13 hours"
# future
assert timesince(then_future, now) == "0 minutes"
def test_timesince():
then = datetime(2010, 4, 12, 12, 30, 0)
then_timestamp = 1271075400.0
then_future = datetime(2012, 4, 12, 12, 30, 0)
now = datetime(2010, 5, 15, 1, 50, 0)
now_timestamp = 1273888200.0
# timestamp
assert timesince(then_timestamp, now_timestamp) == "1 month and 2 days"
# basic
assert timesince(then, now) == "1 month and 2 days"
# count
assert timesince(then, now, count=3) == "1 month, 2 days and 13 hours"
# future
assert timesince(then_future, now) == "0 minutes"
def tellinput(event, conn, db, nick, notice):
"""
:type event: cloudbot.event.Event
:type conn: cloudbot.client.Client
:type db: sqlalchemy.orm.Session
"""
if 'showtells' in event.content.lower():
return
tells = get_unread(db, conn.name, nick)
if tells:
user_from, message, time_sent = tells[0]
reltime = timeformat.timesince(time_sent)
if reltime == 0:
reltime_formatted = "just a moment"
else:
reltime_formatted = reltime
reply = "{} sent you a message {} ago: {}".format(user_from, reltime_formatted, message)
if len(tells) > 1:
reply += " (+{} more, {}showtells to view)".format(len(tells) - 1, conn.config["command_prefix"])
read_tell(db, conn.name, nick, message)
notice(reply)
def tellinput(event, conn, db, nick, notice):
"""
:type event: cloudbot.event.Event
:type conn: cloudbot.client.Client
:type db: sqlalchemy.orm.Session
"""
if 'showtells' in event.content.lower():
return
tells = get_unread(db, conn.name, nick)
if tells:
user_from, message, time_sent = tells[0]
reltime = timeformat.timesince(time_sent)
if reltime == 0:
reltime_formatted = "just a moment"
else:
reltime_formatted = reltime
reply = "{} sent you a message {} ago: {}".format(user_from, reltime_formatted, message)
if len(tells) > 1:
reply += " (+{} more, {}showtells to view)".format(len(tells) - 1, conn.config["command_prefix"])
read_tell(db, conn.name, nick, message)
notice(reply)
def twitter_url(match):
# Find the tweet ID from the URL
tweet_id = match.group(1)
# Get the tweet using the tweepy API
if tw_api is None:
return
try:
tweet = tw_api.get_status(tweet_id)
user = tweet.user
except tweepy.error.TweepError:
return
# Format the return the text of the tweet
text = " ".join(tweet.text.split())
if user.verified:
prefix = "\u2713"
else:
prefix = ""
time = timeformat.timesince(tweet.created_at, datetime.utcnow())
return "{}@\x02{}\x02 ({}): {} ({} ago)".format(prefix, user.screen_name, user.name, text, time)
def pre(text):
"""pre <query> -- searches scene releases using orlydb.com"""
try:
request = requests.get("http://orlydb.com/", params={"q": text})
request.raise_for_status()
except requests.exceptions.HTTPError as e:
return 'Unable to fetch results: {}'.format(e)
h = html.fromstring(request.text)
results = h.xpath("//div[@id='releases']/div/span[@class='release']/..")
if not results:
return "No results found."
result = results[0]
date = result.xpath("span[@class='timestamp']/text()")[0]
section = result.xpath("span[@class='section']//text()")[0]
name = result.xpath("span[@class='release']/text()")[0]
# parse date/time
date = datetime.datetime.strptime(date, "%Y-%m-%d %H:%M:%S")
date_string = date.strftime("%d %b %Y")
since = timeformat.timesince(date)
size = result.xpath("span[@class='inforight']//text()")
if size:
size = ' - ' + size[0].split()[0]
else:
size = ''
return '{} - {}{} - {} ({} ago)'.format(section, name, size, date_string,
since)
def seen(text, nick, chan, db, event, conn):
"""<nick> <channel> - tells when a nickname was last in active in one of my channels
:type db: sqlalchemy.orm.Session
:type event: cloudbot.event.Event
:type conn: cloudbot.client.Client
"""
if event.conn.nick.lower() == text.lower():
return "You need to get your eyes checked."
if text.lower() == nick.lower():
return "Have you looked in a mirror lately?"
if not re.match("^[A-Za-z0-9_|.\-\]\[]*$", text.lower()):
return "I can't look up that name, its impossible to use!"
db_init(db, conn.name)
last_seen = db.execute("select name, time, quote from seen_user where name like :name and chan = :chan",
{'name': text, 'chan': chan}).fetchone()
if last_seen:
reltime = timeformat.timesince(last_seen[1])
if last_seen[0] != text.lower(): # for glob matching
text = last_seen[0]
if last_seen[2][0:1] == "\x01":
return '{} was last seen {} ago: * {} {}'.format(text, reltime, text, last_seen[2][8:-1])
else:
return '{} was last seen {} ago saying: {}'.format(text, reltime, last_seen[2])
else:
return "I've never seen {} talking in this channel.".format(text)
def pre(text):
"""pre <query> -- searches scene releases using orlydb.com"""
try:
request = requests.get("http://orlydb.com/", params={"q": text})
request.raise_for_status()
except requests.exceptions.HTTPError as e:
return 'Unable to fetch results: {}'.format(e)
h = html.fromstring(request.text)
results = h.xpath("//div[@id='releases']/div/span[@class='release']/..")
if not results:
return "No results found."
result = results[0]
date = result.xpath("span[@class='timestamp']/text()")[0]
section = result.xpath("span[@class='section']//text()")[0]
name = result.xpath("span[@class='release']/text()")[0]
# parse date/time
date = datetime.datetime.strptime(date, "%Y-%m-%d %H:%M:%S")
date_string = date.strftime("%d %b %Y")
since = timeformat.timesince(date)
size = result.xpath("span[@class='inforight']//text()")
if size:
size = ' - ' + size[0].split()[0]
else:
size = ''
return '{} - {}{} - {} ({} ago)'.format(section, name, size, date_string, since)
def twitter_url(match):
# Find the tweet ID from the URL
tweet_id = match.group(1)
# Get the tweet using the tweepy API
if tw_api is None:
return
try:
tweet = tw_api.get_status(tweet_id)
user = tweet.user
except tweepy.error.TweepError:
return
# Format the return the text of the tweet
text = " ".join(tweet.text.split())
if user.verified:
prefix = "\u2713"
else:
prefix = ""
time = timeformat.timesince(tweet.created_at, datetime.utcnow())
return "{}@\x02{}\x02 ({}): {} ({} ago)".format(prefix, user.screen_name,
user.name, text, time)
def showtells(nick, notice, db, conn):
"""showtells -- View all pending tell messages (sent in a notice)."""
tells = get_unread(db, conn.name, nick)
if not tells:
notice("You have no pending messages.")
return
for tell in tells:
sender, message, time_sent = tell
past = timeformat.timesince(time_sent)
notice("{} sent you a message {} ago: {}".format(sender, past, message))
read_all_tells(db, conn.name, nick)
def showtells(nick, notice, db, conn):
"""showtells -- View all pending tell messages (sent in a notice)."""
tells = get_unread(db, conn.name, nick)
if not tells:
notice("You have no pending messages.")
return
for tell in tells:
sender, message, time_sent = tell
past = timeformat.timesince(time_sent)
notice("{} sent you a message {} ago: {}".format(sender, past, message))
read_all_tells(db, conn.name, nick)
def format_output(item, show_url=False):
""" takes a reddit post and returns a formatted sting """
item["title"] = formatting.truncate(item["title"], 50)
item["link"] = short_url.format(item["id"])
raw_time = datetime.fromtimestamp(int(item["created_utc"]))
item["timesince"] = timeformat.timesince(raw_time, count=1)
item["comments"] = formatting.pluralize(item["num_comments"], 'comment')
item["points"] = formatting.pluralize(item["score"], 'point')
if item["over_18"]:
item["warning"] = " \x02NSFW\x02"
else:
item["warning"] = ""
if show_url:
return "\x02{title} : {subreddit}\x02 - posted by \x02{author}\x02" \
" {timesince} ago - {comments}, {points} -" \
" {link}{warning}".format(**item)
else:
return "\x02{title} : {subreddit}\x02 - posted by \x02{author}\x02" \
" {timesince} ago - {comments}, {points}{warning}".format(**item)
def seen(text, nick, chan, db, event, conn):
"""<nick> <channel> - tells when a nickname was last in active in one of my channels
:type db: sqlalchemy.orm.Session
:type event: cloudbot.event.Event
:type conn: cloudbot.client.Client
"""
if event.conn.nick.lower() == text.lower():
return "You need to get your eyes checked."
if text.lower() == nick.lower():
return "Have you looked in a mirror lately?"
if not re.match("^[A-Za-z0-9_|.\-\]\[]*$", text.lower()):
return "I can't look up that name, its impossible to use!"
db_init(db, conn.name)
last_seen = db.execute(
"select name, time, quote from seen_user where name like :name and chan = :chan",
{
'name': text,
'chan': chan
}).fetchone()
if last_seen:
reltime = timeformat.timesince(last_seen[1])
if last_seen[0] != text.lower(): # for glob matching
text = last_seen[0]
if last_seen[2][0:1] == "\x01":
return '{} was edast seen {} ago: * {} {}'.format(
text, reltime, text, last_seen[2][8:-1])
else:
return '{} was last seen {} ago saying: {}'.format(
text, reltime, last_seen[2])
else:
return "I've never seen {} talking in this channel.".format(text)
def twitter(text):
"""twitter <user> [n] -- Gets last/[n]th tweet from <user>"""
if tw_api is None:
return
if re.match(r'^\d+$', text):
# user is getting a tweet by id
try:
# get tweet by id
tweet = tw_api.get_status(text)
except tweepy.error.TweepError as e:
if e[0][0]['code'] == 34:
return "Could not find tweet."
else:
return "Error {}: {}".format(e[0][0]['code'], e[0][0]['message'])
user = tweet.user
elif re.match(r'^\w{1,15}$', text) or re.match(r'^\w{1,15}\s+\d+$', text):
# user is getting a tweet by name
if text.find(' ') == -1:
username = text
tweet_number = 0
else:
username, tweet_number = text.split()
tweet_number = int(tweet_number) - 1
if tweet_number > 300:
return "This command can only find the last \x02300\x02 tweets."
try:
# try to get user by username
user = tw_api.get_user(username)
except tweepy.error.TweepError as e:
if e[0][0]['code'] == 34:
return "Could not find user."
else:
return "Error {}: {}".format(e[0][0]['code'], e[0][0]['message'])
# get the users tweets
user_timeline = tw_api.user_timeline(id=user.id, count=tweet_number + 1)
# if the timeline is empty, return an error
if not user_timeline:
return "The user \x02{}\x02 has no tweets.".format(user.screen_name)
# grab the newest tweet from the users timeline
try:
tweet = user_timeline[tweet_number]
except IndexError:
tweet_count = len(user_timeline)
return "The user \x02{}\x02 only has \x02{}\x02 tweets.".format(user.screen_name, tweet_count)
elif re.match(r'^#\w+$', text):
# user is searching by hashtag
search = tw_api.search(text)
if not search:
return "No tweets found."
tweet = random.choice(search)
user = tweet.user
else:
# ???
return "Invalid Input"
# Format the return the text of the tweet
text = " ".join(tweet.text.split())
if user.verified:
prefix = "\u2713"
else:
prefix = ""
time = timeformat.timesince(tweet.created_at, datetime.utcnow())
return "{}@\x02{}\x02 ({}): {} ({} ago)".format(prefix, user.screen_name, user.name, text, time)
def twitter(text):
"""twitter <user> [n] -- Gets last/[n]th tweet from <user>"""
if tw_api is None:
return
if re.match(r'^\d+$', text):
# user is getting a tweet by id
try:
# get tweet by id
tweet = tw_api.get_status(text)
except tweepy.error.TweepError as e:
if e[0][0]['code'] == 34:
return "Could not find tweet."
else:
return "Error {}: {}".format(e[0][0]['code'],
e[0][0]['message'])
user = tweet.user
elif re.match(r'^\w{1,15}$', text) or re.match(r'^\w{1,15}\s+\d+$', text):
# user is getting a tweet by name
if text.find(' ') == -1:
username = text
tweet_number = 0
else:
username, tweet_number = text.split()
tweet_number = int(tweet_number) - 1
if tweet_number > 300:
return "This command can only find the last \x02300\x02 tweets."
try:
# try to get user by username
user = tw_api.get_user(username)
except tweepy.error.TweepError as e:
if e[0][0]['code'] == 34:
return "Could not find user."
else:
return "Error {}: {}".format(e[0][0]['code'],
e[0][0]['message'])
# get the users tweets
user_timeline = tw_api.user_timeline(id=user.id,
count=tweet_number + 1)
# if the timeline is empty, return an error
if not user_timeline:
return "The user \x02{}\x02 has no tweets.".format(
user.screen_name)
# grab the newest tweet from the users timeline
try:
tweet = user_timeline[tweet_number]
except IndexError:
tweet_count = len(user_timeline)
return "The user \x02{}\x02 only has \x02{}\x02 tweets.".format(
user.screen_name, tweet_count)
elif re.match(r'^#\w+$', text):
# user is searching by hashtag
search = tw_api.search(text)
if not search:
return "No tweets found."
tweet = random.choice(search)
user = tweet.user
else:
# ???
return "Invalid Input"
# Format the return the text of the tweet
text = " ".join(tweet.text.split())
if user.verified:
prefix = "\u2713"
else:
prefix = ""
time = timeformat.timesince(tweet.created_at, datetime.utcnow())
return "{}@\x02{}\x02 ({}): {} ({} ago)".format(prefix, user.screen_name,
user.name, text, time)
def lastfm(text, nick, db, bot, notice):
"""[user] [dontsave] - displays the now playing (or last played) track of LastFM user [user]"""
api_key = bot.config.get("api_keys", {}).get("lastfm")
if not api_key:
return "error: no api key set"
# check if the user asked us not to save his details
dontsave = text.endswith(" dontsave")
if dontsave:
user = text[:-9].strip().lower()
else:
user = text
db.execute("create table if not exists lastfm(nick primary key, acc)")
if not user:
user = db.execute("select acc from lastfm where nick=lower(:nick)",
{'nick': nick}).fetchone()
if not user:
notice(lastfm.__doc__)
return
user = user[0]
params = {'method': 'user.getrecenttracks', 'api_key': api_key, 'user': user, 'limit': 1}
request = requests.get(api_url, params=params)
response = request.json()
if 'error' in response:
return "Error: {}.".format(response["message"])
if "track" not in response["recenttracks"] or len(response["recenttracks"]["track"]) == 0:
return 'No recent tracks for user "{}" found.'.format(user)
tracks = response["recenttracks"]["track"]
if type(tracks) == list:
# if the user is listening to something, the tracks entry is a list
# the first item is the current track
track = tracks[0]
status = 'is listening to'
ending = '.'
elif type(tracks) == dict:
# otherwise, they aren't listening to anything right now, and
# the tracks entry is a dict representing the most recent track
track = tracks
status = 'last listened to'
# lets see how long ago they listened to it
time_listened = datetime.fromtimestamp(int(track["date"]["uts"]))
time_since = timeformat.timesince(time_listened)
ending = ' ({} ago)'.format(time_since)
else:
return "error: could not parse track listing"
title = track["name"]
album = track["album"]["#text"]
artist = track["artist"]["#text"]
out = '{} {} "{}"'.format(user, status, title)
if artist:
out += " by \x02{}\x0f".format(artist)
if album:
out += " from the album \x02{}\x0f".format(album)
# append ending based on what type it was
out += ending
if text and not dontsave:
db.execute("insert or replace into lastfm(nick, acc) values (:nick, :account)",
{'nick': nick.lower(), 'account': user})
db.commit()
return out
def lastfm(text, nick, db, bot, notice):
"""[user] [dontsave] - displays the now playing (or last played) track of LastFM user [user]"""
api_key = bot.config.get("api_keys", {}).get("lastfm")
if not api_key:
return "No last.fm API key set."
# check if the user asked us not to save his details
dontsave = text.endswith(" dontsave")
if dontsave:
user = text[:-9].strip().lower()
else:
user = text
db.execute("create table if not exists lastfm(nick primary key, acc)")
if not user:
user = db.execute("select acc from lastfm where nick=lower(:nick)", {
'nick': nick
}).fetchone()
if not user:
notice(lastfm.__doc__)
return
user = user[0]
params = {
'method': 'user.getrecenttracks',
'api_key': api_key,
'user': user,
'limit': 1
}
request = requests.get(api_url, params=params)
if request.status_code != requests.codes.ok:
return "Failed to fetch info ({})".format(request.status_code)
response = request.json()
if 'error' in response:
return "Last.FM Error: {}.".format(response["message"])
if "track" not in response["recenttracks"] or len(
response["recenttracks"]["track"]) == 0:
return 'No recent tracks for user "{}" found.'.format(user)
tracks = response["recenttracks"]["track"]
if type(tracks) == list:
# if the user is listening to something, the tracks entry is a list
# the first item is the current track
track = tracks[0]
status = 'is listening to'
ending = '.'
elif type(tracks) == dict:
# otherwise, they aren't listening to anything right now, and
# the tracks entry is a dict representing the most recent track
track = tracks
status = 'last listened to'
# lets see how long ago they listened to it
time_listened = datetime.fromtimestamp(int(track["date"]["uts"]))
time_since = timeformat.timesince(time_listened)
ending = ' ({} ago)'.format(time_since)
else:
return "error: could not parse track listing"
title = track["name"]
album = track["album"]["#text"]
artist = track["artist"]["#text"]
out = '{} {} "{}"'.format(user, status, title)
if artist:
out += " by \x02{}\x0f".format(artist)
if album:
out += " from the album \x02{}\x0f".format(album)
# append ending based on what type it was
out += ending
if text and not dontsave:
db.execute(
"insert or replace into lastfm(nick, acc) values (:nick, :account)",
{
'nick': nick.lower(),
'account': user
})
db.commit()
return out