mirror of
https://github.com/lucaspalomodevelop/eventcally.git
synced 2026-03-13 08:09:37 +00:00
103 lines
2.7 KiB
Python
103 lines
2.7 KiB
Python
import json
|
|
import os
|
|
import shutil
|
|
|
|
import click
|
|
from flask.cli import AppGroup
|
|
from sqlalchemy import and_
|
|
from sqlalchemy.orm import joinedload
|
|
|
|
from project import app, dump_path
|
|
from project.api.event.schemas import EventDumpSchema
|
|
from project.api.event_category.schemas import EventCategoryDumpSchema
|
|
from project.api.event_reference.schemas import EventReferenceDumpSchema
|
|
from project.api.organization.schemas import OrganizationDumpSchema
|
|
from project.api.organizer.schemas import OrganizerDumpSchema
|
|
from project.api.place.schemas import PlaceDumpSchema
|
|
from project.models import (
|
|
AdminUnit,
|
|
Event,
|
|
EventCategory,
|
|
EventOrganizer,
|
|
EventPlace,
|
|
EventReference,
|
|
PublicStatus,
|
|
)
|
|
from project.utils import make_dir
|
|
|
|
dump_cli = AppGroup("dump")
|
|
|
|
|
|
def dump_items(items, schema, file_base_name, dump_path):
|
|
result = schema.dump(items)
|
|
path = os.path.join(dump_path, file_base_name + ".json")
|
|
|
|
with open(path, "w") as outfile:
|
|
json.dump(result, outfile, ensure_ascii=False)
|
|
|
|
click.echo(f"{len(items)} item(s) dumped to {path}.")
|
|
|
|
|
|
@dump_cli.command("all")
|
|
def dump_all():
|
|
# Setup temp dir
|
|
tmp_path = os.path.join(dump_path, "tmp")
|
|
make_dir(tmp_path)
|
|
|
|
# Events
|
|
events = (
|
|
Event.query.join(Event.admin_unit)
|
|
.options(joinedload(Event.categories))
|
|
.filter(
|
|
and_(
|
|
Event.public_status == PublicStatus.published,
|
|
AdminUnit.is_verified,
|
|
)
|
|
)
|
|
.all()
|
|
)
|
|
dump_items(events, EventDumpSchema(many=True), "events", tmp_path)
|
|
|
|
# Places
|
|
places = EventPlace.query.all()
|
|
dump_items(places, PlaceDumpSchema(many=True), "places", tmp_path)
|
|
|
|
# Event categories
|
|
event_categories = EventCategory.query.all()
|
|
dump_items(
|
|
event_categories,
|
|
EventCategoryDumpSchema(many=True),
|
|
"event_categories",
|
|
tmp_path,
|
|
)
|
|
|
|
# Organizers
|
|
organizers = EventOrganizer.query.all()
|
|
dump_items(organizers, OrganizerDumpSchema(many=True), "organizers", tmp_path)
|
|
|
|
# Organizations
|
|
organizations = AdminUnit.query.all()
|
|
dump_items(
|
|
organizations, OrganizationDumpSchema(many=True), "organizations", tmp_path
|
|
)
|
|
|
|
# Event references
|
|
event_references = EventReference.query.all()
|
|
dump_items(
|
|
event_references,
|
|
EventReferenceDumpSchema(many=True),
|
|
"event_references",
|
|
tmp_path,
|
|
)
|
|
|
|
# Zip
|
|
zip_base_name = os.path.join(dump_path, "all")
|
|
zip_path = shutil.make_archive(zip_base_name, "zip", tmp_path)
|
|
click.echo(f"Zipped all up to {zip_path}.")
|
|
|
|
# Clean up temp dir
|
|
shutil.rmtree(tmp_path, ignore_errors=True)
|
|
|
|
|
|
app.cli.add_command(dump_cli)
|