bpexport.py

#!/usr/bin/env python3
# -*- coding: utf8 -*-
################################################################################
##
## This file is part of BetterPonymotes.
## Copyright (c) 2012-2015 Typhos.
##
## 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.
##
## You should have received a copy of the GNU Affero General Public License
## along with this program.  If not, see <http://www.gnu.org/licenses/>.
##
################################################################################

import argparse

import bplib
import bplib.json
import bplib.objects
import bplib.resolve

def dump_json(context, emotes):
    data = {}
    for (name, emote) in emotes.items():
        if emote.source.variant_matches is None:
            emote.source.match_variants()
        info = encode(emote, context)
        data[name] = info
    return data

def encode(emote, context):
    # Info: is_nsfw?, source, size, tags
    base = emote.base_variant()
    root = emote.source.variant_matches[emote]
    all_tags = root.all_tags(context) | emote.all_tags(context)

    for tag in list(all_tags):
        all_tags |= set(context.tag_config["TagAliases"].get(tag, []))

    is_nsfw = "+nsfw" in all_tags
    size = max(base.size) if hasattr(base, "size") else 0
    emitted_tags = [tag for tag in all_tags if tag not in context.tag_config["HiddenTags"]]

    info = {"source": emote.source.name, "tags": emitted_tags, "size": size}
    if is_nsfw:
        info["is_nsfw"] = is_nsfw
    if base.css:
        info["css"] = base.css
    if hasattr(base, "image_url"): # FIXME
        info["image_url"] = bplib.image_download_url(base.image_url)
        info["size"] = base.size
        info["offset"] = base.offset
    if emote.name != root.name: # Redirect
        info["primary"] = root.name
    return info

def main():
    parser = argparse.ArgumentParser(description="Export processed emote data")
    parser.add_argument("-js", "--json", help="Output file", default="export.json")
    args = parser.parse_args()

    print("Loading emotes")
    context = bplib.objects.Context()
    context.load_config()
    context.load_sources()

    print("Processing")
    emotes, all_emotes = bplib.resolve.resolve_emotes(context)

    data = dump_json(context, emotes)

    print("Dumping")
    with open(args.json, "w") as file:
        bplib.json.dump(data, file, indent=0, max_depth=1, sort_keys=True)

if __name__ == "__main__":
    main()