mathdatech/tootbot
1
0
Fork 0
mirror of https://github.com/cquest/tootbot.git synced 2024-11-30 15:21:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

pep8

Browse Source
This commit is contained in:
cquest 2019-04-03 17:46:13 +02:00
parent 720aff1abd
commit 184ab026f4
1 changed files with 15 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
tootbot.py
View File

@ -9,20 +9,21 @@ from mastodon import Mastodon
import requests import requests
if len(sys.argv) < 4: if len(sys.argv) < 4:
print("Usage: python3 tootbot.py twitter_account mastodon_login mastodon_passwd mastodon_instance") print("Usage: python3 tootbot.py twitter_account mastodon_login mastodon_passwd mastodon_instance") # noqa
sys.exit(1) sys.exit(1)
# sqlite db to store processed tweets (and corresponding toots ids) # sqlite db to store processed tweets (and corresponding toots ids)
sql = sqlite3.connect('tootbot.db') sql = sqlite3.connect('tootbot.db')
db = sql.cursor() db = sql.cursor()
db.execute('''CREATE TABLE IF NOT EXISTS tweets (tweet text, toot text, twitter text, mastodon text, instance text)''') db.execute('''CREATE TABLE IF NOT EXISTS tweets (tweet text, toot text,
twitter text, mastodon text, instance text)''')
if len(sys.argv)>4: if len(sys.argv) > 4:
instance = sys.argv[4] instance = sys.argv[4]
else: else:
instance = 'amicale.net' instance = 'amicale.net'
if len(sys.argv)>5: if len(sys.argv) > 5:
days = int(sys.argv[5]) days = int(sys.argv[5])
else: else:
days = 1 days = 1
@ -37,7 +38,7 @@ d = feedparser.parse('http://twitrss.me/twitter_user_to_rss/?user='+twitter)
for t in reversed(d.entries): for t in reversed(d.entries):
# check if this tweet has been processed # check if this tweet has been processed
db.execute('SELECT * FROM tweets WHERE tweet = ? AND twitter = ? and mastodon = ? and instance = ?',(t.id, twitter, mastodon, instance)) db.execute('SELECT * FROM tweets WHERE tweet = ? AND twitter = ? and mastodon = ? and instance = ?', (t.id, source, mastodon, instance)) # noqa
last = db.fetchone() last = db.fetchone()
# process only unprocessed tweets less than 1 day old # process only unprocessed tweets less than 1 day old
@ -48,7 +49,7 @@ for t in reversed(d.entries):
if Mastodon.create_app( if Mastodon.create_app(
'tootbot', 'tootbot',
api_base_url='https://'+instance, api_base_url='https://'+instance,
to_file = instance+'.secret' to_file=instance+'.secret'
): ):
print('tootbot app created on instance '+instance) print('tootbot app created on instance '+instance)
else: else:
@ -86,20 +87,24 @@ for t in reversed(d.entries):
if m != None: if m != None:
l = m.group(0) l = m.group(0)
r = requests.get(l, allow_redirects=False) r = requests.get(l, allow_redirects=False)
if r.status_code in {301,302}: if r.status_code in {301, 302}:
c = c.replace(l,r.headers.get('Location')) c = c.replace(l, r.headers.get('Location'))
# remove pic.twitter.com links # remove pic.twitter.com links
m = re.search(r"pic.twitter.com[^ \xa0]*", c) m = re.search(r"pic.twitter.com[^ \xa0]*", c)
if m != None: if m != None:
l = m.group(0) l = m.group(0)
c = c.replace(l,' ') c = c.replace(l, ' ')
# remove ellipsis # remove ellipsis
c = c.replace('\xa0…',' ') c = c.replace('\xa0…',' ')
if toot_media is not None: if toot_media is not None:
toot = mastodon_api.status_post(c, in_reply_to_id=None, media_ids=toot_media, sensitive=False, visibility='public', spoiler_text=None) toot = mastodon_api.status_post(c, in_reply_to_id=None,
media_ids=toot_media,
sensitive=False,
visibility='public',
spoiler_text=None)
if "id" in toot: if "id" in toot:
db.execute("INSERT INTO tweets VALUES ( ? , ? , ? , ? , ? )", db.execute("INSERT INTO tweets VALUES ( ? , ? , ? , ? , ? )",
(t.id, toot["id"], twitter, mastodon, instance)) (t.id, toot["id"], twitter, mastodon, instance))