89 lines
3.3 KiB
Python
Executable File
89 lines
3.3 KiB
Python
Executable File
#!/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 main.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)
|