def testWrite(self):
api = Api(BLOG, USER, PASSWORD)
newpost = api.write_regular('title', 'body')
post = api.read(newpost['id'])
assert newpost['id'] == post['id']
newpost = api.write_link('http://www.google.com')
post = api.read(newpost['id'])
assert newpost['id'] == post['id']
newpost = api.write_quote('it was the best of times...')
post = api.read(newpost['id'])
assert newpost['id'] == post['id']
newpost = api.write_conversation('me: wow\nyou: double wow!')
post = api.read(newpost['id'])
assert newpost['id'] == post['id']
newpost = api.write_video('http://www.youtube.com/watch?v=60og9gwKh1o')
post = api.read(newpost['id'])
assert newpost['id'] == post['id']
newpost = api.write_photo(
'http://www.google.com/intl/en_ALL/images/logo.gif')
post = api.read(newpost['id'])
assert newpost['id'] == post['id']
def testBadAuthenticate(self):
api = Api(BLOG, USER, 'badpassword')
try:
api.auth_check()
assert False # should never get here
except TumblrAuthError, e:
pass
def testRequiredArgs(self):
api = Api(BLOG, USER, PASSWORD)
self.assertRaises(TumblrError, api.write_regular)
self.assertRaises(TumblrError, api.write_quote)
self.assertRaises(TumblrError, api.write_photo)
self.assertRaises(TumblrError,
api.write_photo,
source='foo',
data='bar')
self.assertRaises(TumblrError, api.write_conversation)
self.assertRaises(TumblrError, api.write_link)
self.assertRaises(TumblrError, api.write_video)
def testFixNames(self):
api = Api(BLOG)
before = {}
before['test_one'] = 1
before['test_two'] = 1
after = api._fixnames(before)
assert not 'test_one' in after
assert 'test-one' in after
assert 'test-two' in after
assert not 'test_two' in after
def handle_url(self, message, reply_to, url, sender, times=0):
new_args = message.strip().split(' ')
tag_args = filter(
lambda arg: len(arg) > 0 and arg[0] == '#'
,new_args
)
real_args = filter(
lambda arg: len(arg) > 0 and arg[0] != '#'
,new_args
)
filtered_msg = ' '.join(real_args).replace(url,'').replace(':','')
times = times + 1
print "Attempting to post %s for the #%i time" % (url, times)
nick = nm_to_n(sender)
api = Api(self.tumblog, self.email, self.password)
caption = '%s\nvia %s' % (filtered_msg, nick)
def print_result(result):
print result
print type(result)
try:
print "%s posted a %s to %s" % (nick, result['type'], result['url'])
except:
print "Call result: %s" % repr(result)
try:
try:
urls = api.readurls()
if url not in api.readurls():
self.bot.set_callback(api.autopost_url, print_result,
args=(url, caption, tag_args))
except BrowserStateError:
# Skipping because of a mechanize error.
return
except URLError:
return
except TumblrError, e:
return
if (times < 3):
print e
print "Error encountered, trying it again."
# try it again, a couple of times.
sleep(3)
self.handle_url(self, message, reply_to, url, sender, times=times)
else:
print e
def testRead(self):
api = Api(BLOG)
freq = {}
posts = api.read()
total = 0
for post in posts:
total += 1
type = post['type']
try:
freq[type] += 1
except:
freq[type] = 1
assert total > 0
for type in freq:
assert self.countType(api, type) == freq[type]
def do_quote(self, message, reply_to):
new_args = message.strip().split(' ')
user = new_args[0]
tag_args = filter(
lambda arg: len(arg) > 0 and arg[0] == '#'
,new_args
)
real_args = filter(
lambda arg: len(arg) > 0 and arg[0] != '#'
,new_args
)
# remove hash
tag_args = [arg[1:] for arg in tag_args]
print "real args: %s\ntag args: %s" % (real_args, tag_args)
if len(real_args) == 1:
# quote a single person
last_said = self.bot.log[reply_to][user][-1].strip()
if last_said != None:
try:
api = Api(self.tumblog, self.email, self.password)
kwargs = {
'body': cgi.escape(last_said),
'tags': tag_args,
'slug': 'quote'
}
def print_result(result):
try:
print "Quoted %s as %s to %s" % (user,
result['url'], repr(reply_to))
except:
print 'Could not quote: %s' % result
self.bot.set_callback(api.write_regular,
print_result, kwargs=kwargs)
#bot.say(channel, "Your quote: %s" % post['url'])
except TumblrError, e:
self.bot.connection.privmsg(reply_to, "Tumblr posting " + \
"failed due to a temporary error.")
print TumblrError, e
except Exception, e:
#bot.say(channel, "Something horrible happened.")
print e
def testAuthenticate(self):
api = Api(BLOG, USER, PASSWORD)
api.auth_check()
