blob: 22e733fa74aa1542fdda44708bfa051d661b0889 (
plain)
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
# Copyright (c) 2023 Florian Fischer. All rights reserved.
#
# This file is part of geldschieberbot.
#
# geldschieberbot is free software: you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option)
# any later version.
#
# geldschieberbot 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License along with
# geldschieberbot found in the LICENSE file. If not,
# see <https://www.gnu.org/licenses/>.
"""Models used by geldschieberbot"""
import json
import typing as T
from dataclasses import dataclass, field
from utils import to_euro
@dataclass
class MessageContext:
"""Class representing the context of a message passed to a command function"""
sender_number: str
sender: T.Optional[str]
args: list[str]
body: list[str]
timestamp: str
@dataclass
class Modification:
"""Class representing a single modification to the balance
Amount is transfered from donor to the recipient.
"""
recipient: str
donor: str
amount: int
def in_string(self) -> str:
"""Format the change using the recipient as initiator"""
return (f'{self.recipient} {"->" if self.amount < 0 else "<-"}'
f' {to_euro(abs(self.amount))} {self.donor}')
def out_string(self) -> str:
"""Format the change using the donor as initiator"""
return (f'{self.donor} {"->" if self.amount < 0 else "<-"}'
f' {to_euro(abs(self.amount))} {self.recipient}')
@dataclass
class Change:
"""Class representing a change to the state caused by a single command"""
cmd: list[str]
modifications: list[Modification]
timestamp: str
rewind_cmds: list[list[str]] = field(default_factory=lambda: [])
@dataclass
class Quote:
"""Class representing a message to quote"""
timestamp: str
author: str
class GeldschieberbotJSONEncoder(json.JSONEncoder):
"""Custom JSONEncoder supporting our dataclasses"""
def default(self, o):
if isinstance(o, (Modification, Change)):
return o.__dict__
return json.JSONEncoder.default(self, o)
@dataclass
class Reply:
"""Class representing a reply from the bot"""
msg: str
attachment: T.Optional[str] = None
|
