Skip to content

models/expensedb.py

models.expensedb

ExpenseDB

ExpenseDB

Bases: DBBase

Expense Collection

Source code in models/expensedb.py 
 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
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
class ExpenseDB(DBBase):
    ''' Expense Collection '''

    def __init__(self) -> None:
        super().__init__('expense')

    def index(self) -> None:
        ''' To make collection's index

        Indexs:
            - `pid`, `tid`
            - `pid`, `tid`, `request.buid`
            - `pid`, `tid`, `create_by`

        '''
        self.create_index([('pid', 1), ('tid', 1)])
        self.create_index([('pid', 1), ('tid', 1), ('request.buid', 1)])
        self.create_index([('pid', 1), ('tid', 1), ('create_by', 1)])

    @staticmethod
    def status() -> dict[str, str]:
        ''' Status mapping

        Returns:
            `1`: `已申請`, `2`: `審核中`, `3`: `出款中`, `4`: `已出款`, `5`: `已完成`

        '''
        return {'1': '已申請',
                '2': '審核中',
                '3': '出款中',
                '4': '已出款',
                '5': '已完成',
                }

    @staticmethod
    def new(pid: str, tid: str, uid: str) -> dict[str, Any]:
        ''' Create new data object

        Args:
            pid (str): project id
            tid (str): team id
            uid (str): user id

        Struct:
            - ``_id``: Unique expense id.
            - ``pid``: Project id.
            - ``tid``: Team id.
            - ``dispense_id``: Reference to expense DB, if any
            - ``request``: This object include the budget data in
                           `buid`, `desc`, `paydate`, `code`.
            - ``invoices``: List of invoice data, in `iv_id`,
                            `currency`, `name`, `status`, `total`, `received`.
            - ``bank``: User's bank account info, the data from user's
                        real_profile in settings.
            - ``status``: The mappings from [`ExpenseDB.status`][models.expensedb.ExpenseDB.status].
            - ``note``: Note fields, but not implememnt in expense page.
            - ``code``: Random shorter code for reference in batch.
            - ``releveant_code``: The codes are releveant to others.
            - ``create_by``: Create by who in `uid`.
            - ``create_at``: When to created.
            - ``enable`` (`bool`): enable or not.

        Returns:
            Return an base data object in `dict`.

        TODO:
            Need refactor in pydantic.

        '''
        return {
            '_id': f'{uuid4().node:x}',
            'pid': pid,
            'tid': tid,
            'dispense_id': '',
            'request': {},
            'invoices': [],
            'bank': {},
            'status': '1',  # 已申請/處理中/已出款/已完成
            'note': {'myself': '', 'to_create': ''},
            'code': f"E-{''.join(choices(string.ascii_uppercase+string.digits, k=4))}",
            'relevant_code': [],
            'create_by': uid,
            'create_at': datetime.today(),
            'enable': True,
        }

    def add(self, data: dict[str, Any]) -> dict[str, Any]:
        ''' Add data, using `pid`, `tid`, `_id` as the key to insert / update.

        Args:
            data (dict): the data to insert / update.

        Returns:
            Return the inserted / updated data.

        '''
        return self.find_one_and_update(
            {'pid': data['pid'], 'tid': data['tid'], '_id': data['_id']},
            {'$set': data},
            upsert=True,
            return_document=ReturnDocument.AFTER,
        )

add 

add(data: dict[str, Any]) -> dict[str, Any]

Add data, using pid, tid, _id as the key to insert / update.

Parameters:

Name Type Description Default
data dict

the data to insert / update.

 required

Returns:

Type Description
dict[str, Any]

Return the inserted / updated data.
Source code in models/expensedb.py 
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
def add(self, data: dict[str, Any]) -> dict[str, Any]:
    ''' Add data, using `pid`, `tid`, `_id` as the key to insert / update.

    Args:
        data (dict): the data to insert / update.

    Returns:
        Return the inserted / updated data.

    '''
    return self.find_one_and_update(
        {'pid': data['pid'], 'tid': data['tid'], '_id': data['_id']},
        {'$set': data},
        upsert=True,
        return_document=ReturnDocument.AFTER,
    )

index 

index() -> None

To make collection's index

Indexs
  • pid, tid
  • pid, tid, request.buid
  • pid, tid, create_by
Source code in models/expensedb.py 
19
20
21
22
23
24
25
26
27
28
29
30
def index(self) -> None:
    ''' To make collection's index

    Indexs:
        - `pid`, `tid`
        - `pid`, `tid`, `request.buid`
        - `pid`, `tid`, `create_by`

    '''
    self.create_index([('pid', 1), ('tid', 1)])
    self.create_index([('pid', 1), ('tid', 1), ('request.buid', 1)])
    self.create_index([('pid', 1), ('tid', 1), ('create_by', 1)])

new staticmethod 

new(pid: str, tid: str, uid: str) -> dict[str, Any]

Create new data object

Parameters:

Name Type Description Default
pid str

project id

 required
tid str

team id

 required
uid str

user id

 required
Struct
  • _id: Unique expense id.
  • pid: Project id.
  • tid: Team id.
  • dispense_id: Reference to expense DB, if any
  • request: This object include the budget data in buid, desc, paydate, code.
  • invoices: List of invoice data, in iv_id, currency, name, status, total, received.
  • bank: User's bank account info, the data from user's real_profile in settings.
  • status: The mappings from ExpenseDB.status.
  • note: Note fields, but not implememnt in expense page.
  • code: Random shorter code for reference in batch.
  • releveant_code: The codes are releveant to others.
  • create_by: Create by who in uid.
  • create_at: When to created.
  • enable (bool): enable or not.

Returns:

Type Description
dict[str, Any]

Return an base data object in dict.
TODO

Need refactor in pydantic.

Source code in models/expensedb.py 
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
@staticmethod
def new(pid: str, tid: str, uid: str) -> dict[str, Any]:
    ''' Create new data object

    Args:
        pid (str): project id
        tid (str): team id
        uid (str): user id

    Struct:
        - ``_id``: Unique expense id.
        - ``pid``: Project id.
        - ``tid``: Team id.
        - ``dispense_id``: Reference to expense DB, if any
        - ``request``: This object include the budget data in
                       `buid`, `desc`, `paydate`, `code`.
        - ``invoices``: List of invoice data, in `iv_id`,
                        `currency`, `name`, `status`, `total`, `received`.
        - ``bank``: User's bank account info, the data from user's
                    real_profile in settings.
        - ``status``: The mappings from [`ExpenseDB.status`][models.expensedb.ExpenseDB.status].
        - ``note``: Note fields, but not implememnt in expense page.
        - ``code``: Random shorter code for reference in batch.
        - ``releveant_code``: The codes are releveant to others.
        - ``create_by``: Create by who in `uid`.
        - ``create_at``: When to created.
        - ``enable`` (`bool`): enable or not.

    Returns:
        Return an base data object in `dict`.

    TODO:
        Need refactor in pydantic.

    '''
    return {
        '_id': f'{uuid4().node:x}',
        'pid': pid,
        'tid': tid,
        'dispense_id': '',
        'request': {},
        'invoices': [],
        'bank': {},
        'status': '1',  # 已申請/處理中/已出款/已完成
        'note': {'myself': '', 'to_create': ''},
        'code': f"E-{''.join(choices(string.ascii_uppercase+string.digits, k=4))}",
        'relevant_code': [],
        'create_by': uid,
        'create_at': datetime.today(),
        'enable': True,
    }

status staticmethod 

status() -> dict[str, str]

Status mapping

Returns:

Type Description
dict[str, str]

1: 已申請, 2: 審核中, 3: 出款中, 4: 已出款, 5: 已完成
Source code in models/expensedb.py 
32
33
34
35
36
37
38
39
40
41
42
43
44
45
@staticmethod
def status() -> dict[str, str]:
    ''' Status mapping

    Returns:
        `1`: `已申請`, `2`: `審核中`, `3`: `出款中`, `4`: `已出款`, `5`: `已完成`

    '''
    return {'1': '已申請',
            '2': '審核中',
            '3': '出款中',
            '4': '已出款',
            '5': '已完成',
            }