__init__.py
2.21 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
# -*- coding: utf8 -*-
# This file is part of PyBossa.
#
# Copyright (C) 2015 SciFabric LTD.
#
# PyBossa 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.
#
# PyBossa 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 PyBossa. If not, see <http://www.gnu.org/licenses/>.
from pybossa.core import db
import factory
from pybossa.repositories import UserRepository
from pybossa.repositories import ProjectRepository
from pybossa.repositories import BlogRepository
from pybossa.repositories import TaskRepository
from pybossa.repositories import AuditlogRepository
from pybossa.repositories import WebhookRepository
user_repo = UserRepository(db)
project_repo = ProjectRepository(db)
blog_repo = BlogRepository(db)
task_repo = TaskRepository(db)
auditlog_repo = AuditlogRepository(db)
webhook_repo = WebhookRepository(db)
def reset_all_pk_sequences():
ProjectFactory.reset_sequence()
BlogpostFactory.reset_sequence()
CategoryFactory.reset_sequence()
TaskFactory.reset_sequence()
TaskRunFactory.reset_sequence()
UserFactory.reset_sequence()
AuditlogFactory.reset_sequence()
WebhookFactory.reset_sequence()
class BaseFactory(factory.Factory):
@classmethod
def _setup_next_sequence(cls):
return 1
@classmethod
def _build(cls, model_class, *args, **kwargs):
project = model_class(*args, **kwargs)
db.session.remove()
return project
# Import the factories
from project_factory import ProjectFactory
from blogpost_factory import BlogpostFactory
from category_factory import CategoryFactory
from task_factory import TaskFactory
from taskrun_factory import TaskRunFactory, AnonymousTaskRunFactory
from user_factory import UserFactory
from auditlog_factory import AuditlogFactory
from webhook_factory import WebhookFactory