1
|
|
|
""" |
2
|
|
|
byceps.services.email.service |
3
|
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ |
4
|
|
|
|
5
|
|
|
:Copyright: 2006-2021 Jochen Kupperschmidt |
6
|
|
|
:License: Revised BSD (see `LICENSE` file for details) |
7
|
|
|
""" |
8
|
|
|
|
9
|
1 |
|
from __future__ import annotations |
10
|
1 |
|
from email.utils import parseaddr |
11
|
1 |
|
from typing import Optional |
12
|
|
|
|
13
|
1 |
|
from sqlalchemy.exc import IntegrityError |
14
|
|
|
|
15
|
1 |
|
from ...database import db, upsert |
16
|
1 |
|
from ... import email |
17
|
1 |
|
from ...typing import BrandID |
18
|
1 |
|
from ...util.jobqueue import enqueue |
19
|
|
|
|
20
|
1 |
|
from .dbmodels import EmailConfig as DbEmailConfig |
21
|
1 |
|
from .transfer.models import EmailConfig, Message, NameAndAddress |
22
|
|
|
|
23
|
|
|
|
24
|
1 |
|
class UnknownEmailConfigId(ValueError): |
25
|
1 |
|
pass |
26
|
|
|
|
27
|
|
|
|
28
|
1 |
|
def create_config( |
29
|
|
|
brand_id: BrandID, |
30
|
|
|
sender_address: str, |
31
|
|
|
*, |
32
|
|
|
sender_name: Optional[str] = None, |
33
|
|
|
contact_address: Optional[str] = None, |
34
|
|
|
) -> EmailConfig: |
35
|
|
|
"""Create a configuration.""" |
36
|
1 |
|
config = DbEmailConfig( |
37
|
|
|
brand_id, |
38
|
|
|
sender_address, |
39
|
|
|
sender_name=sender_name, |
40
|
|
|
contact_address=contact_address, |
41
|
|
|
) |
42
|
|
|
|
43
|
1 |
|
db.session.add(config) |
44
|
1 |
|
db.session.commit() |
45
|
|
|
|
46
|
1 |
|
return _db_entity_to_config(config) |
47
|
|
|
|
48
|
|
|
|
49
|
1 |
|
def update_config( |
50
|
|
|
brand_id: BrandID, |
51
|
|
|
sender_address: str, |
52
|
|
|
sender_name: Optional[str], |
53
|
|
|
contact_address: Optional[str], |
54
|
|
|
) -> EmailConfig: |
55
|
|
|
"""Update a configuration.""" |
56
|
|
|
config = _find_db_config(brand_id) |
57
|
|
|
|
58
|
|
|
if config is None: |
59
|
|
|
raise UnknownEmailConfigId( |
60
|
|
|
f'No e-mail config found for brand ID "{brand_id}"' |
61
|
|
|
) |
62
|
|
|
|
63
|
|
|
config.sender_address = sender_address |
64
|
|
|
config.sender_name = sender_name |
65
|
|
|
config.contact_address = contact_address |
66
|
|
|
|
67
|
|
|
db.session.commit() |
68
|
|
|
|
69
|
|
|
return _db_entity_to_config(config) |
70
|
|
|
|
71
|
|
|
|
72
|
1 |
|
def delete_config(brand_id: BrandID) -> bool: |
73
|
|
|
"""Delete a configuration. |
74
|
|
|
|
75
|
|
|
It is expected that no database records (sites) refer to the |
76
|
|
|
configuration anymore. |
77
|
|
|
|
78
|
|
|
Return `True` on success, or `False` if an error occurred. |
79
|
|
|
""" |
80
|
1 |
|
get_config(brand_id) # Verify ID exists. |
81
|
|
|
|
82
|
1 |
|
try: |
83
|
1 |
|
db.session \ |
84
|
|
|
.query(DbEmailConfig) \ |
85
|
|
|
.filter_by(brand_id=brand_id) \ |
86
|
|
|
.delete() |
87
|
|
|
|
88
|
1 |
|
db.session.commit() |
89
|
|
|
except IntegrityError: |
90
|
|
|
db.session.rollback() |
91
|
|
|
return False |
92
|
|
|
|
93
|
1 |
|
return True |
94
|
|
|
|
95
|
|
|
|
96
|
1 |
|
def _find_db_config(brand_id: BrandID) -> Optional[DbEmailConfig]: |
97
|
1 |
|
return db.session \ |
98
|
|
|
.query(DbEmailConfig) \ |
99
|
|
|
.filter_by(brand_id=brand_id) \ |
100
|
|
|
.one_or_none() |
101
|
|
|
|
102
|
|
|
|
103
|
1 |
|
def get_config(brand_id: BrandID) -> EmailConfig: |
104
|
|
|
"""Return the configuration, or raise an error if none is configured |
105
|
|
|
for that brand. |
106
|
|
|
""" |
107
|
1 |
|
config = _find_db_config(brand_id) |
108
|
|
|
|
109
|
1 |
|
if config is None: |
110
|
|
|
raise UnknownEmailConfigId( |
111
|
|
|
f'No e-mail config found for brand ID "{brand_id}"' |
112
|
|
|
) |
113
|
|
|
|
114
|
1 |
|
return _db_entity_to_config(config) |
115
|
|
|
|
116
|
|
|
|
117
|
1 |
|
def set_config( |
118
|
|
|
brand_id: BrandID, |
119
|
|
|
sender_address: str, |
120
|
|
|
*, |
121
|
|
|
sender_name: Optional[str] = None, |
122
|
|
|
contact_address: Optional[str] = None, |
123
|
|
|
) -> None: |
124
|
|
|
"""Add or update configuration for that ID.""" |
125
|
1 |
|
table = DbEmailConfig.__table__ |
126
|
1 |
|
identifier = { |
127
|
|
|
'brand_id': brand_id, |
128
|
|
|
'sender_address': sender_address, |
129
|
|
|
} |
130
|
1 |
|
replacement = { |
131
|
|
|
'sender_name': sender_name, |
132
|
|
|
'contact_address': contact_address, |
133
|
|
|
} |
134
|
|
|
|
135
|
1 |
|
upsert(table, identifier, replacement) |
136
|
|
|
|
137
|
|
|
|
138
|
1 |
|
def get_all_configs() -> list[EmailConfig]: |
139
|
|
|
"""Return all configurations.""" |
140
|
|
|
configs = db.session.query(DbEmailConfig).all() |
141
|
|
|
|
142
|
|
|
return [_db_entity_to_config(config) for config in configs] |
143
|
|
|
|
144
|
|
|
|
145
|
1 |
|
def parse_address(address_str: str) -> NameAndAddress: |
146
|
|
|
"""Parse a string into name and address parts.""" |
147
|
|
|
name, address = parseaddr(address_str) |
148
|
|
|
|
149
|
|
|
if not name and not address: |
150
|
|
|
raise ValueError(f'Could not parse name and address value: "{address}"') |
151
|
|
|
|
152
|
|
|
return NameAndAddress(name, address) |
153
|
|
|
|
154
|
|
|
|
155
|
1 |
|
def enqueue_message(message: Message) -> None: |
156
|
|
|
"""Enqueue e-mail to be sent asynchronously.""" |
157
|
1 |
|
enqueue_email( |
158
|
|
|
message.sender, message.recipients, message.subject, message.body |
159
|
|
|
) |
160
|
|
|
|
161
|
|
|
|
162
|
1 |
|
def enqueue_email( |
163
|
|
|
sender: NameAndAddress, |
164
|
|
|
recipients: list[str], |
165
|
|
|
subject: str, |
166
|
|
|
body: str, |
167
|
|
|
) -> None: |
168
|
|
|
"""Enqueue e-mail to be sent asynchronously.""" |
169
|
1 |
|
sender_str = sender.format() |
170
|
1 |
|
enqueue(send_email, sender_str, recipients, subject, body) |
171
|
|
|
|
172
|
|
|
|
173
|
1 |
|
def send_email( |
174
|
|
|
sender: str, recipients: list[str], subject: str, body: str |
175
|
|
|
) -> None: |
176
|
|
|
"""Send e-mail.""" |
177
|
1 |
|
email.send(sender, recipients, subject, body) |
178
|
|
|
|
179
|
|
|
|
180
|
1 |
|
def _db_entity_to_config(config: DbEmailConfig) -> EmailConfig: |
181
|
1 |
|
sender = NameAndAddress( |
182
|
|
|
name=config.sender_name, |
183
|
|
|
address=config.sender_address, |
184
|
|
|
) |
185
|
|
|
|
186
|
1 |
|
return EmailConfig( |
187
|
|
|
brand_id=config.brand_id, |
188
|
|
|
sender=sender, |
189
|
|
|
contact_address=config.contact_address, |
190
|
|
|
) |
191
|
|
|
|