Python create_category 예제들

예제 #1

파일: test_tag_categories.py 프로젝트: rr-/szurubooru

def test_create_category_when_first():
    with patch('szurubooru.func.tag_categories.update_category_name'), \
            patch('szurubooru.func.tag_categories.update_category_color'):
        category = tag_categories.create_category('name', 'color')
        assert category.default
        tag_categories.update_category_name \
            .assert_called_once_with(category, 'name')
        tag_categories.update_category_color \
            .assert_called_once_with(category, 'color')

예제 #2

파일: test_tag_categories.py 프로젝트: devtest8/szurubooru

def test_create_category_when_first():
    with patch('szurubooru.func.tag_categories.update_category_name'), \
            patch('szurubooru.func.tag_categories.update_category_color'):
        category = tag_categories.create_category('name', 'color')
        assert category.default
        tag_categories.update_category_name \
            .assert_called_once_with(category, 'name')
        tag_categories.update_category_color \
            .assert_called_once_with(category, 'color')

예제 #3

파일: tags.py 프로젝트: hnamquoc/szurubooru

def update_tag_category_name(tag, category_name):
    category = db.session \
        .query(db.TagCategory) \
        .filter(db.TagCategory.name == category_name) \
        .first()
    if not category:
        category = tag_categories.create_category(
            category_name, DEFAULT_CATEGORY_COLOR)
        db.session.add(category)
    tag.category = category

예제 #4

def create_tag_category(ctx: rest.Context,
                        _params: Dict[str, str] = {}) -> rest.Response:
    auth.verify_privilege(ctx.user, "tag_categories:create")
    name = ctx.get_param_as_string("name")
    color = ctx.get_param_as_string("color")
    category = tag_categories.create_category(name, color)
    ctx.session.add(category)
    ctx.session.flush()
    snapshots.create(category, ctx.user)
    ctx.session.commit()
    return _serialize(ctx, category)

예제 #5

파일: tag_category_api.py 프로젝트: princessGasmask/szurubooru

 def post(self, ctx):
     auth.verify_privilege(ctx.user, 'tag_categories:create')
     name = ctx.get_param_as_string('name', required=True)
     color = ctx.get_param_as_string('color', required=True)
     category = tag_categories.create_category(name, color)
     ctx.session.add(category)
     ctx.session.flush()
     snapshots.create(category, ctx.user)
     ctx.session.commit()
     tags.export_to_json()
     return _serialize_category_with_details(category)

예제 #6

파일: tag_category_api.py 프로젝트: hnamquoc/szurubooru

 def post(self, ctx):
     auth.verify_privilege(ctx.user, 'tag_categories:create')
     name = ctx.get_param_as_string('name', required=True)
     color = ctx.get_param_as_string('color', required=True)
     category = tag_categories.create_category(name, color)
     ctx.session.add(category)
     ctx.session.flush()
     snapshots.save_entity_creation(category, ctx.user)
     ctx.session.commit()
     tags.export_to_json()
     return tag_categories.serialize_category_with_details(category)

예제 #7

파일: test_tag_categories.py 프로젝트: rr-/szurubooru

def test_create_category_when_subsequent(tag_category_factory):
    db.session.add(tag_category_factory())
    db.session.flush()
    with patch('szurubooru.func.tag_categories.update_category_name'), \
            patch('szurubooru.func.tag_categories.update_category_color'):
        category = tag_categories.create_category('name', 'color')
        assert not category.default
        tag_categories.update_category_name \
            .assert_called_once_with(category, 'name')
        tag_categories.update_category_color \
            .assert_called_once_with(category, 'color')

예제 #8

파일: test_tag_categories.py 프로젝트: devtest8/szurubooru

def test_create_category_when_subsequent(tag_category_factory):
    db.session.add(tag_category_factory())
    db.session.flush()
    with patch('szurubooru.func.tag_categories.update_category_name'), \
            patch('szurubooru.func.tag_categories.update_category_color'):
        category = tag_categories.create_category('name', 'color')
        assert not category.default
        tag_categories.update_category_name \
            .assert_called_once_with(category, 'name')
        tag_categories.update_category_color \
            .assert_called_once_with(category, 'color')

예제 #9

파일: tag_category_api.py 프로젝트: rr-/szurubooru

def create_tag_category(
        ctx: rest.Context, _params: Dict[str, str] = {}) -> rest.Response:
    auth.verify_privilege(ctx.user, 'tag_categories:create')
    name = ctx.get_param_as_string('name')
    color = ctx.get_param_as_string('color')
    category = tag_categories.create_category(name, color)
    ctx.session.add(category)
    ctx.session.flush()
    snapshots.create(category, ctx.user)
    ctx.session.commit()
    return _serialize(ctx, category)

예제 #10

def test_create_category_when_first():
    with patch("szurubooru.func.tag_categories.update_category_name"), patch(
            "szurubooru.func.tag_categories.update_category_color"), patch(
                "szurubooru.func.tag_categories.update_category_order"):
        category = tag_categories.create_category("name", "color", 7)
        assert category.default
        tag_categories.update_category_name.assert_called_once_with(
            category, "name")
        tag_categories.update_category_color.assert_called_once_with(
            category, "color")
        tag_categories.update_category_order.assert_called_once_with(
            category, 7)

예제 #11

def test_create_category_when_subsequent(tag_category_factory):
    db.session.add(tag_category_factory())
    db.session.flush()
    with patch("szurubooru.func.tag_categories.update_category_name"), patch(
            "szurubooru.func.tag_categories.update_category_color"), patch(
                "szurubooru.func.tag_categories.update_category_order"):
        category = tag_categories.create_category("name", "color", 7)
        assert not category.default
        tag_categories.update_category_name.assert_called_once_with(
            category, "name")
        tag_categories.update_category_color.assert_called_once_with(
            category, "color")
        tag_categories.update_category_order.assert_called_once_with(
            category, 7)