diff --git a/packages/plugins/@hera/plugin-rental/src/server/services/record-service.ts b/packages/plugins/@hera/plugin-rental/src/server/services/record-service.ts index 11cf4ad3a..3bda5ac09 100644 --- a/packages/plugins/@hera/plugin-rental/src/server/services/record-service.ts +++ b/packages/plugins/@hera/plugin-rental/src/server/services/record-service.ts @@ -1,7 +1,14 @@ import Database, { CreateOptions, MagicAttributeModel, Transaction } from '@tachybase/database'; import { Db, Service } from '@tachybase/utils'; -import { ConversionLogics, Movement, RecordCategory, RecordTypes, settlementStatus } from '../../utils/constants'; +import { + ConversionLogics, + Movement, + ProjectCategory, + RecordCategory, + RecordTypes, + settlementStatus, +} from '../../utils/constants'; import validateLicensePlate from '../../utils/validateLIcensePlate'; @Service() @@ -160,7 +167,12 @@ export class RecordService { const associatedCompanyProject = await this.db .getRepository('project') .findOne({ where: { company_id: out_stock.company_id } }); - in_stock = associatedCompanyProject; + // + if (out_stock.category !== ProjectCategory.companyWarehouse) { + in_stock = associatedCompanyProject; + } else { + in_stock = values.in_stock; + } } else if (values.category === RecordCategory.staging) { in_stock = values.in_stock; out_stock = values.out_stock; diff --git a/packages/plugins/@hera/plugin-rental/src/utils/constants.ts b/packages/plugins/@hera/plugin-rental/src/utils/constants.ts index 8f4a4201a..ce51faa25 100644 --- a/packages/plugins/@hera/plugin-rental/src/utils/constants.ts +++ b/packages/plugins/@hera/plugin-rental/src/utils/constants.ts @@ -234,4 +234,19 @@ export enum RecordTypes { rentOutStock = '5', } +export enum ProjectCategory { + /**1:公司仓库 */ + companyWarehouse = '1', + /** + * 2:项目仓库 + */ + projectWarehouse = '2', + /**3:租赁客户 */ + rentalCustomers = '3', + /**4:同行客户 */ + peerCustomers = '4', + /**5:供应商 */ + supplier = '5', +} + export const RulesNumber = 99999;