def test_get_next_id_with_existing_one(self): from psi.app.utils import db_util from psi.app.models import ProductCategory with self.test_client: fixture.login_as_admin(self.test_client) object_faker.category() next_id = db_util.get_next_id(ProductCategory) self.assertEquals(3, next_id)
def test_get_next_id_with_existing_one(self): from psi.app.utils import db_util from psi.app.models import ProductCategory with self.test_client: fixture.login_as_admin(self.test_client) object_faker.category() next_id = db_util.get_next_id(ProductCategory) self.assertEquals(3, next_id)
def category(self, category_id=None, creator=current_user): from psi.app.models import ProductCategory category = ProductCategory() category.id = category_id if category_id is not None else db_util.get_next_id( ProductCategory) category.code = self.faker.pystr(max_chars=8) category.name = self.faker.name() category.organization = creator.organization return category
def purchase_order(self, po_id=None, number_of_line=1, creator=current_user, type=None, status=None): from psi.app.models import PurchaseOrder, PurchaseOrderLine, EnumValues po = PurchaseOrder() po.remark = self.faker.text(max_nb_chars=20) po.logistic_amount = self.faker.pyfloat(positive=True, left_digits=2, right_digits=0) po.order_date = self.faker.date() if status is None: draft_status = EnumValues.get(const.PO_DRAFT_STATUS_KEY) po.status = draft_status else: po.status = status po.status_id = status.id if type is None: types = EnumValues.type_filter(const.PO_TYPE_KEY).all() type = random.choice(types) if type.code == const.FRANCHISE_PO_TYPE_KEY: if creator.organization.parent is not None: po.to_organization = creator.organization.parent else: po.to_organization = creator.organization po.type = type po.type_id = type.id po.id = po_id if po_id is not None else db_util.get_next_id( PurchaseOrder) po.organization = creator.organization po.supplier = self.supplier(creator=creator) for i in range(0, number_of_line): line = PurchaseOrderLine() line.remark = self.faker.text(max_nb_chars=10) line.id = db_util.get_next_id(PurchaseOrderLine) line.product = self.product(supplier=po.supplier, creator=creator) line.purchase_order = po line.quantity = random.randint(1, 100) line.unit_price = self.faker.pydecimal(positive=True, left_digits=3, right_digits=0) return po
def sales_order(self, so_id=None, number_of_line=1, creator=current_user, type=None, products=None): from psi.app.models import SalesOrder, EnumValues, SalesOrderLine from psi.app.const import SO_CREATED_STATUS_KEY from psi.app.services import SalesOrderService so = SalesOrder() so.remark = self.faker.text(max_nb_chars=20) so.logistic_amount = self.faker.pyfloat(positive=True, left_digits=2, right_digits=0) so.customer = self.customer(creator=creator) so.order_date = self.faker.date() created_status = EnumValues.get(SO_CREATED_STATUS_KEY) so.status_id = created_status.id so.organization = creator.organization if type is None: types = EnumValues.type_filter(const.SO_TYPE_KEY).all() so.type = random.choice(types) else: so.type = type so.id = so_id if so_id is not None else db_util.get_next_id(SalesOrder) for i in range(0, number_of_line): line = SalesOrderLine() line.remark = self.faker.text(max_nb_chars=10) line.id = db_util.get_next_id(SalesOrderLine) if products is None: line.product = self.product(creator=creator) else: line.product = products[i] line.sales_order = so line.quantity = random.randint(1, 100) line.unit_price = self.faker.pydecimal(positive=True, left_digits=3, right_digits=0) SalesOrderService.create_or_update_incoming(so) SalesOrderService.create_or_update_expense(so) return so
def supplier(self, supplier_id=None, creator=current_user): from psi.app.models import Supplier supplier = Supplier() supplier.id = supplier_id if supplier_id is not None else db_util.get_next_id( Supplier) supplier.name = self.faker.name() supplier.can_mixed_whole_sale = self.faker.pybool() supplier.contact = self.faker.name() supplier.email = self.faker.email() supplier.phone = self.faker.phone_number() supplier.qq = self.faker.pyint() supplier.organization = creator.organization supplier.remark = self.faker.pystr(max_chars=100) supplier.website = self.faker.uri()[:64] return supplier
def product(self, product_id=None, supplier=None, creator=current_user): from psi.app.models import Product product = Product() product.id = product_id if product_id is not None else db_util.get_next_id( Product) product.category = self.category(creator=creator) product.name = self.faker.name() product.deliver_day = random.randint(3, 7) product.supplier = self.supplier( creator=creator) if supplier is None else supplier product.distinguishing_feature = self.faker.paragraphs(nb=3) product.lead_day = random.randint(1, 5) product.need_advice = self.faker.pybool() product.purchase_price = random.randint(1, 100) product.retail_price = product.purchase_price + random.randint(1, 100) product.organization = creator.organization return product
def customer(self, customer_id=None, creator=current_user): from psi.app.models import Customer customer = Customer() customer.id = customer_id if customer_id is not None else db_util.get_next_id( Customer) customer.address = self.faker.address() customer.birthday = self.faker.date_time_this_decade() customer.email = self.faker.safe_email() customer.first_name = self.faker.first_name() customer.last_name = self.faker.last_name() customer.join_date = self.faker.date_time_this_decade() customer.join_channel = random.choice( customer.join_channel_filter().all()) customer.level = random.choice(customer.level_filter().all()) customer.organization = creator.organization customer.points = self.faker.pyint() return customer
def organization(self, organization_id=None, type=None, parent=None): from psi.app.models import Organization from psi.app.models import EnumValues organization = Organization() organization.description = self.faker.text(max_nb_chars=10) organization.name = self.faker.name() result = db_util.get_result_raw_sql( "select max(lft), max(rgt) from organization") organization.lft = result[0] + 1 organization.rgt = result[1] + 1 organization.id = organization_id if organization_id is not None else db_util.get_next_id( Organization) if type is None: organization.type = EnumValues.get(u"DIRECT_SELLING_STORE") else: organization.type = type if parent is not None: organization.parent = parent return organization
def create_so_from_fpo(purchase_order): so_type = EnumValues.get(const.FRANCHISE_SO_TYPE_KEY) sales_order = SalesOrder() sales_order.id = db_util.get_next_id(SalesOrder) sales_order.order_date = purchase_order.order_date sales_order.type = so_type sales_order.remark = "PO ID: [{0}]".format(purchase_order.id) sales_order.organization = purchase_order.to_organization sales_order.status = EnumValues.get(const.SO_CREATED_STATUS_KEY) lines = [] for line in purchase_order.lines: sol = SalesOrderLine() sol.unit_price = line.unit_price sol.quantity = line.quantity sol.sales_order = sales_order sol.product = line.product sol.remark = "PO Line ID:[{0}]".format(line.id) lines.append(sol) from psi.app.services import SalesOrderService incoming = SalesOrderService.create_or_update_incoming(sales_order) expense = SalesOrderService.create_or_update_expense(sales_order) return sales_order, incoming, expense
def create_so_from_fpo(purchase_order): so_type = EnumValues.get(const.FRANCHISE_SO_TYPE_KEY) sales_order = SalesOrder() sales_order.id = db_util.get_next_id(SalesOrder) sales_order.order_date = purchase_order.order_date sales_order.type = so_type sales_order.remark = "PO ID: [{0}]".format(purchase_order.id) sales_order.organization = purchase_order.to_organization sales_order.status = EnumValues.get(const.SO_CREATED_STATUS_KEY) lines = [] for line in purchase_order.lines: sol = SalesOrderLine() sol.unit_price = line.unit_price sol.quantity = line.quantity sol.sales_order = sales_order sol.product = line.product sol.remark = "PO Line ID:[{0}]".format(line.id) lines.append(sol) from psi.app.services import SalesOrderService incoming = SalesOrderService.create_or_update_incoming(sales_order) expense = SalesOrderService.create_or_update_expense(sales_order) return sales_order, incoming, expense
def test_get_next_id(self): from psi.app.utils import db_util from psi.app.models import Supplier next_id = db_util.get_next_id(Supplier) self.assertEquals(1, next_id)
def test_get_next_id(self): from psi.app.utils import db_util from psi.app.models import Supplier next_id = db_util.get_next_id(Supplier) self.assertEquals(1, next_id)