import click
from flask.cli import FlaskGroup, ScriptInfo
from flask import Flask
from flask_restful import Api, Resource, marshal_with, marshal_with_field, fields

from . import model as m

class Tag(Resource):
    tag_fields = {
            'name' : fields.String,
            'prefix' : fields.String
            }

class TagList(Resource):
    tag_list = { fields.List(fields.Nested(Tag.tag_fields)) }

    @marshal_with_field(fields.List(fields.Nested(Tag.tag_fields)))
    def get(self):
        res = list(m.Tag.select().where(~m.Tag.default).dicts().iterator())
        print(res)
        return res

def create_app(info):
    app = Flask('archivist')
    api = Api(app)
    api.add_resource(TagList, '/')
    return app

server_group = FlaskGroup(
        name='server',
        context_settings = {'obj' : ScriptInfo(create_app=create_app)})