properly split text for tweets and toots

ttadmin/common/social.py

@@ -1,3 +1,5 @@
+import re
+
 from mastodon import Mastodon
 import tweepy
 
@@ -14,41 +16,49 @@ tw_auth = tweepy.OAuthHandler(settings.TWITTER_CONSUMER_KEY, settings.TWITTER_CO
 tw_auth.set_access_token(settings.TWITTER_TOKEN, settings.TWITTER_TOKEN_SECRET)
 twitter = tweepy.API(tw_auth)
 
-def post_to_mastodon(qs):
+def split_posts_by_length(text, length):
+    pattern = '.{,%d}(?:\s|$)' % length - 1
+    chunks = re.findall(pattern, text)
     posts = []
-    if len(qs) > 1:
+    post = ''
-        welcome = 'Welcome new user '
+    for chunk in chunks:
+        if len(post + chunk) <= length:
+            post += chunk
         else:
-        welcome  = 'Welcome new users!!!\n\n'
+            posts.append(post)
-    message = welcome
+            post = chunk
-    for townie in qs:
+    return posts
-        if len(message + townie.username) + 1 > 500:
+
-            posts.append(message.strip())
+
-            message = welcome + '~{}\n'.format(townie.username)
+def post_to_mastodon(message):
-        else:
+    posts = split_posts_by_length(message, 500)
-            message += '~{}\n'.format(townie.username)
+    status_info = None
-    posts.append(message.strip())
     for post in posts:
-        mastodon.post(post)
+        if status_info:
+            status_info = mastodon.status_post(post, in_reply_to_id=status_info['id'])
+        else:
+            status_info = mastodon.status_post(post)
 
-def post_to_twitter(qs):
+
-    posts = []
+def post_to_twitter(message):
-    if len(qs) > 1:
+    posts = split_posts_by_length(message, 140)
-        welcome = 'Welcome new user '
+    status_info = None
-    else:
-        welcome = 'Welcome new users!!!\n\n'
-    message = welcome
-    for townie in qs:
-        if len(message + townie.username) + 1 > 140:
-            posts.append(message.strip())
-            message = welcome + '~{}\n'.format(townie.username)
-        else:
-            message += '~{}\n'.format(townie.username)
-    posts.append(message.strip())
     for post in posts:
-        twitter.update_status(post)
+        if status_info:
+            status_info = twitter.update_status(post, in_reply_to_status_id=status_info.id)
+        else:
+            status_info = twitter.update_status(post)
 
-def post_to_social(qs):
+
-    post_to_twitter(qs)
+def post_users_to_social(qs):
-    post_to_mastodon(qs)
+    users = ''
+    for townie in qs:
+        users += '~{}\n'.format(townie.username)
+    users = users.strip()
+    if len(qs) > 1:
+        message = 'Welcome new users!!!\n\n{}'.format(users)
+    else:
+        message = 'Welcome new user {}!'.format(users)
+    post_to_mastodon(message)
+    post_to_twitter(message)