rct-guest/post.py

78 lines
2.8 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):
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":
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)
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)
thought = random.choice(data['thoughts'][thought_type])
if thought_type == "plain":
pass
elif thought_type == "item-stall":
objects = get_object(thought_type)
thought = thought.format(objects[0], objects[1])
else:
thought = thought.format(get_object(thought_type))
print(thought)