You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

89 lines
3.3 KiB

#!/usr/bin/env python3
# https://github.com/Lynnesbian/fediverse-bot-template
# Copyright (C) 2019 Lynne (@lynnesbian@fedi.lynnesbian.space)
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published
# by the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
# import Mastodon.py and other modules
from mastodon import Mastodon
import json, random
try:
cfg = json.load(open('config.json', 'r'))
except:
print("Couldn't load config.json. Make sure you run login.py first!\n-----")
raise
meta = json.load(open('meta.json', 'r'))
# log in
client = Mastodon(
client_id=cfg['client']['id'],
client_secret=cfg['client']['secret'],
access_token=cfg['secret'],
api_base_url=cfg['site'])
# make a post!
data = json.load(open('data.json', 'r'))
def get_object(kind):
# fill in the given thought with the necessary objects.
# for example, if the kind is "rides", generate a string like "Ride Name 1".
num = random.choice([1, 1, 1, 1, 1, 2, 2, 3])
if kind == "rides":
return "{} {}".format(random.choice(data['objects']['thrilling-rides'] + data['objects']['gentle-rides']), num)
elif kind in ["gentle-rides", "thrilling-rides"]:
return "{} {}".format(random.choice(data['objects'][kind]), num)
elif kind == "items":
return random.choice(list(data['objects']['stalls']['edible'].values()) + list(data['objects']['stalls']['other'].values()))
elif kind == "items-edible":
return random.choice(list(data['objects']['stalls']['edible'].values()))
elif kind == "items-other":
return random.choice(list(data['objects']['stalls']['other'].values()))
elif kind == "stalls":
return "{} {}".format(random.choice(list(data['objects']['stalls']['edible'].keys()) + list(data['objects']['stalls']['other'].keys())), num)
elif kind == "item-stall":
# return an item and the stall it comes from.
stalls_weighted = []
for key in data['objects']['stalls']:
for i in range(len(data['objects']['stalls'][key])):
stalls_weighted.append(key)
stall_type = random.choice(stalls_weighted)
choice = random.choice(list(data['objects']['stalls'][stall_type].items()))[::-1]
stall = "{} {}".format(choice[1], num)
return (choice[0], stall)
# generate a list of thought types for a weighted random choice.
# for example, if there were two plain thoughts and one rides thought, the list would be:
# ['plain', 'plain', 'rides']
thoughts_weighted = []
for key in data['thoughts']:
for i in range(len(data['thoughts'][key])):
thoughts_weighted.append(key)
thought_type = random.choice(thoughts_weighted)
# choose a random thought of the type chosen above
thought = random.choice(data['thoughts'][thought_type])
if thought_type == "plain":
# we don't need to fill in anything here.
pass
elif thought_type == "item-stall":
# special case, get_object returns a tuple to fill in two blanks
objects = get_object(thought_type)
thought = thought.format(objects[0], objects[1])
else:
# just fill in the one blank
thought = thought.format(get_object(thought_type))
# post the thought
client.status_post(thought)