Merge pull request #24 from notagoat/pin-branch

Add un/pin to the bot

main.py

@@ -11,8 +11,9 @@ import os, sqlite3, signal, sys, json, re, shutil
 import requests
 import functions
 
-scopes = ["read:statuses", "read:accounts", "read:follows", "write:statuses", "read:notifications"]
+scopes = ["read:statuses", "read:accounts", "read:follows", "write:statuses", "read:notifications", "write:accounts"]
 #cfg defaults
+
 cfg = {
 	"site": "https://botsin.space",
 	"cw": None,
@@ -21,10 +22,10 @@ cfg = {
 	"mention_handling": 1,
 	"max_thread_length": 15
 }
+
 try:
 	cfg.update(json.load(open('config.json', 'r')))
 except:
-
 	shutil.copy2("config.sample.json", "config.json")
 	cfg.update(json.load(open('config.json', 'r')))
 

reply.py

@@ -39,6 +39,7 @@ class ReplyListener(mastodon.StreamListener):
 			posts = 0
 			for post in context['ancestors']:
 				if post['account']['id'] == me:
+					pin = post["id"] #Only used if pin is called, but easier to call here
 					posts += 1
 				if posts >= cfg['max_thread_length']:
 					# stop replying
@@ -46,6 +47,27 @@ class ReplyListener(mastodon.StreamListener):
 					return
 
 			mention = extract_toot(notification['status']['content'])
+			if (mention == "pin") or (mention == "unpin"): #check for keywords
+				print("Found pin/unpin")
+				#get a list of people the bot is following
+				validusers = client.account_following(me)
+				for user in validusers:
+					if user["id"] == notification["account"]["id"]: #user is #valid
+						print("User is valid")
+						visibility = notification['status']['visibility']
+						if visibility == "public":
+							visibility = "unlisted"
+						if mention == "pin":
+							print("pin received, pinning")
+							client.status_pin(pin)
+							client.status_post("Toot pinned!", post_id, visibility=visibility, spoiler_text = cfg['cw'])
+						else:
+							print("unpin received, unpinning")
+							client.status_post("Toot unpinned!", post_id, visibility=visibility, spoiler_text = cfg['cw'])
+							client.status_unpin(pin)
+					else:
+						print("User is not valid")
+			else:
 				toot = functions.make_toot(True)['toot'] #generate a toot
 				toot = acct + " " + toot #prepend the @
 				print(acct + " says " + mention) #logging