From a2b6d3d2e045e31ae82e41e36372fefd3bd02e8b Mon Sep 17 00:00:00 2001 From: Klaus-Uwe Mitterer Date: Tue, 13 Sep 2016 20:20:05 +0200 Subject: [PATCH] Allow tweet.py to parse Twitter status URLs for replies --- tweet.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/tweet.py b/tweet.py index 868251e..8fa71c2 100755 --- a/tweet.py +++ b/tweet.py @@ -4,10 +4,12 @@ import argparse, twitools if __name__ == "__main__": parser = argparse.ArgumentParser(description='Send a tweet from the status monitor\'s account.') - parser.add_argument('-r', '--reply', default=None, metavar='ID', type=int, help='reply to tweet ID') + parser.add_argument('-r', '--reply', default=None, metavar='ID', help='reply to tweet ID') parser.add_argument('-t', '--text', default=None, type=str, help='tweet provided string') args = parser.parse_args() + two = twitools.twObject() + try: if not args.text: text = input("> ") @@ -15,4 +17,14 @@ if __name__ == "__main__": text = args.text except KeyboardInterrupt: exit(0) - two.tweet(text, args.reply) + + if isinstance(args.reply, int): + reply = args.reply + else: + try: + if "twitter.com" in args.reply and "status" in args.reply: + reply = int(args.reply.split('/')[-1]) + except: + raise ValueError("Invalid tweet ID passed for -r.") + + two.tweet(text, reply)