notify.py 24.4 KB
Newer Older
1
#!/usr/bin/env python3
2
3

import email.mime.text
4
5
import email.utils
import smtplib
6
7
8
9
import subprocess
import sys
import textwrap

10
11
from sqlalchemy import and_, or_

Lukas Fleischer's avatar
Lukas Fleischer committed
12
13
import aurweb.config
import aurweb.db
14
import aurweb.filters
Lukas Fleischer's avatar
Lukas Fleischer committed
15
import aurweb.l10n
16

17
from aurweb import db, logging
18
19
20
21
22
23
24
from aurweb.models import PackageBase, User
from aurweb.models.package_comaintainer import PackageComaintainer
from aurweb.models.package_comment import PackageComment
from aurweb.models.package_notification import PackageNotification
from aurweb.models.package_request import PackageRequest
from aurweb.models.request_type import RequestType
from aurweb.models.tu_vote import TUVote
25

26
27
logger = logging.get_logger(__name__)

Lukas Fleischer's avatar
Lukas Fleischer committed
28
aur_location = aurweb.config.get('options', 'aur_location')
29
30


31
32
33
def headers_msgid(thread_id):
    return {'Message-ID': thread_id}

34

35
36
37
def headers_reply(thread_id):
    return {'In-Reply-To': thread_id, 'References': thread_id}

38

39
40
41
42
43
44
45
class Notification:
    def get_refs(self):
        return ()

    def get_headers(self):
        return {}

46
47
48
    def get_cc(self):
        return []

Lukas Fleischer's avatar
Lukas Fleischer committed
49
    def get_body_fmt(self, lang):
50
        body = ''
Lukas Fleischer's avatar
Lukas Fleischer committed
51
        for line in self.get_body(lang).splitlines():
52
53
            if line == '--':
                body += '--\n'
54
                continue
55
56
57
            body += textwrap.fill(line, break_long_words=False) + '\n'
        for i, ref in enumerate(self.get_refs()):
            body += '\n' + '[%d] %s' % (i + 1, ref)
Lukas Fleischer's avatar
Lukas Fleischer committed
58
        return body.rstrip()
59

60
    def _send(self) -> None:
61
        sendmail = aurweb.config.get('notifications', 'sendmail')
62
63
        sender = aurweb.config.get('notifications', 'sender')
        reply_to = aurweb.config.get('notifications', 'reply-to')
64
65
66
        reason = self.__class__.__name__
        if reason.endswith('Notification'):
            reason = reason[:-len('Notification')]
67
68

        for recipient in self.get_recipients():
Lukas Fleischer's avatar
Lukas Fleischer committed
69
70
71
72
            to, lang = recipient
            msg = email.mime.text.MIMEText(self.get_body_fmt(lang),
                                           'plain', 'utf-8')
            msg['Subject'] = self.get_subject(lang)
73
74
            msg['From'] = sender
            msg['Reply-to'] = reply_to
Lukas Fleischer's avatar
Lukas Fleischer committed
75
            msg['To'] = to
76
77
            if self.get_cc():
                msg['Cc'] = str.join(', ', self.get_cc())
78
            msg['X-AUR-Reason'] = reason
79
            msg['Date'] = email.utils.formatdate(localtime=True)
80
81
82
83

            for key, value in self.get_headers().items():
                msg[key] = value

84
85
86
87
88
89
90
91
92
93
            sendmail = aurweb.config.get('notifications', 'sendmail')
            if sendmail:
                # send email using the sendmail binary specified in the
                # configuration file
                p = subprocess.Popen([sendmail, '-t', '-oi'],
                                     stdin=subprocess.PIPE)
                p.communicate(msg.as_bytes())
            else:
                # send email using smtplib; no local MTA required
                server_addr = aurweb.config.get('notifications', 'smtp-server')
94
95
96
97
98
99
                server_port = aurweb.config.getint('notifications',
                                                   'smtp-port')
                use_ssl = aurweb.config.getboolean('notifications',
                                                   'smtp-use-ssl')
                use_starttls = aurweb.config.getboolean('notifications',
                                                        'smtp-use-starttls')
100
101
102
                user = aurweb.config.get('notifications', 'smtp-user')
                passwd = aurweb.config.get('notifications', 'smtp-password')

103
104
105
106
107
                classes = {
                    False: smtplib.SMTP,
                    True: smtplib.SMTP_SSL,
                }
                server = classes[use_ssl](server_addr, server_port)
108
109
110
111
112
113
114
115

                if use_starttls:
                    server.ehlo()
                    server.starttls()
                    server.ehlo()

                if user and passwd:
                    server.login(user, passwd)
116
117

                server.set_debuglevel(0)
118
                deliver_to = [to] + self.get_cc()
119
                server.sendmail(sender, deliver_to, msg.as_bytes())
120
                server.quit()
121

122
123
124
125
126
127
128
129
    def send(self) -> None:
        try:
            self._send()
        except OSError as exc:
            logger.error("Unable to emit notification due to an "
                         "OSError (precise exception following).")
            logger.error(str(exc))

130
131

class ResetKeyNotification(Notification):
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
    def __init__(self, uid):

        user = db.query(User).filter(
            and_(User.ID == uid, User.Suspended == 0)
        ).with_entities(
            User.Username,
            User.Email,
            User.BackupEmail,
            User.LangPreference,
            User.ResetKey
        ).order_by(User.Username.asc()).first()

        self._username = user.Username
        self._to = user.Email
        self._backup = user.BackupEmail
        self._lang = user.LangPreference
        self._resetkey = user.ResetKey

Lukas Fleischer's avatar
Lukas Fleischer committed
150
        super().__init__()
151
152

    def get_recipients(self):
153
154
155
156
        if self._backup:
            return [(self._to, self._lang), (self._backup, self._lang)]
        else:
            return [(self._to, self._lang)]
157

Lukas Fleischer's avatar
Lukas Fleischer committed
158
    def get_subject(self, lang):
159
        return aurweb.l10n.translator.translate('AUR Password Reset', lang)
160

Lukas Fleischer's avatar
Lukas Fleischer committed
161
    def get_body(self, lang):
162
163
164
165
166
167
        return aurweb.l10n.translator.translate(
            'A password reset request was submitted for the account '
            '{user} associated with your email address. If you wish to '
            'reset your password follow the link [1] below, otherwise '
            'ignore this message and nothing will happen.',
            lang).format(user=self._username)
168
169
170
171
172
173

    def get_refs(self):
        return (aur_location + '/passreset/?resetkey=' + self._resetkey,)


class WelcomeNotification(ResetKeyNotification):
Lukas Fleischer's avatar
Lukas Fleischer committed
174
    def get_subject(self, lang):
175
176
177
        return aurweb.l10n.translator.translate(
            'Welcome to the Arch User Repository',
            lang)
178

Lukas Fleischer's avatar
Lukas Fleischer committed
179
    def get_body(self, lang):
180
181
182
183
184
        return aurweb.l10n.translator.translate(
            'Welcome to the Arch User Repository! In order to set an '
            'initial password for your new account, please click the '
            'link [1] below. If the link does not work, try copying and '
            'pasting it into your browser.', lang)
185
186
187


class CommentNotification(Notification):
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
    def __init__(self, uid, pkgbase_id, comment_id):

        self._user = db.query(User.Username).filter(
            User.ID == uid).first().Username
        self._pkgbase = db.query(PackageBase.Name).filter(
            PackageBase.ID == pkgbase_id).first().Name

        query = db.query(User).join(PackageNotification).filter(
            and_(User.CommentNotify == 1,
                 PackageNotification.UserID != uid,
                 PackageNotification.PackageBaseID == pkgbase_id,
                 User.Suspended == 0)
        ).with_entities(
            User.Email,
            User.LangPreference
        ).distinct()
        self._recipients = [(u.Email, u.LangPreference) for u in query]

        pkgcomment = db.query(PackageComment.Comments).filter(
            PackageComment.ID == comment_id).first()
        self._text = pkgcomment.Comments

Lukas Fleischer's avatar
Lukas Fleischer committed
210
        super().__init__()
211
212

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
213
        return self._recipients
214

Lukas Fleischer's avatar
Lukas Fleischer committed
215
    def get_subject(self, lang):
216
217
218
        return aurweb.l10n.translator.translate(
            'AUR Comment for {pkgbase}',
            lang).format(pkgbase=self._pkgbase)
219

Lukas Fleischer's avatar
Lukas Fleischer committed
220
    def get_body(self, lang):
221
222
223
        body = aurweb.l10n.translator.translate(
            '{user} [1] added the following comment to {pkgbase} [2]:',
            lang).format(user=self._user, pkgbase=self._pkgbase)
224
        body += '\n\n' + self._text + '\n\n--\n'
225
226
227
228
229
230
        dnlabel = aurweb.l10n.translator.translate(
            'Disable notifications', lang)
        body += aurweb.l10n.translator.translate(
            'If you no longer wish to receive notifications about this '
            'package, please go to the package page [2] and select '
            '"{label}".', lang).format(label=dnlabel)
231
232
233
234
235
236
237
238
239
240
241
242
243
        return body

    def get_refs(self):
        return (aur_location + '/account/' + self._user + '/',
                aur_location + '/pkgbase/' + self._pkgbase + '/')

    def get_headers(self):
        thread_id = '<pkg-notifications-' + self._pkgbase + \
                    '@aur.archlinux.org>'
        return headers_reply(thread_id)


class UpdateNotification(Notification):
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
    def __init__(self, uid, pkgbase_id):

        self._user = db.query(User.Username).filter(
            User.ID == uid).first().Username
        self._pkgbase = db.query(PackageBase.Name).filter(
            PackageBase.ID == pkgbase_id).first().Name

        query = db.query(User).join(PackageNotification).filter(
            and_(User.UpdateNotify == 1,
                 PackageNotification.UserID != uid,
                 PackageNotification.PackageBaseID == pkgbase_id,
                 User.Suspended == 0)
        ).with_entities(
            User.Email,
            User.LangPreference
        ).distinct()
        self._recipients = [(u.Email, u.LangPreference) for u in query]

Lukas Fleischer's avatar
Lukas Fleischer committed
262
        super().__init__()
263
264

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
265
        return self._recipients
266

Lukas Fleischer's avatar
Lukas Fleischer committed
267
    def get_subject(self, lang):
268
269
270
        return aurweb.l10n.translator.translate(
            'AUR Package Update: {pkgbase}',
            lang).format(pkgbase=self._pkgbase)
271

Lukas Fleischer's avatar
Lukas Fleischer committed
272
    def get_body(self, lang):
273
274
275
        body = aurweb.l10n.translator.translate(
            '{user} [1] pushed a new commit to {pkgbase} [2].',
            lang).format(user=self._user, pkgbase=self._pkgbase)
276
        body += '\n\n--\n'
277
278
279
280
281
282
        dnlabel = aurweb.l10n.translator.translate(
            'Disable notifications', lang)
        body += aurweb.l10n.translator.translate(
            'If you no longer wish to receive notifications about this '
            'package, please go to the package page [2] and select '
            '"{label}".', lang).format(label=dnlabel)
283
284
285
286
287
288
289
290
291
292
293
294
295
        return body

    def get_refs(self):
        return (aur_location + '/account/' + self._user + '/',
                aur_location + '/pkgbase/' + self._pkgbase + '/')

    def get_headers(self):
        thread_id = '<pkg-notifications-' + self._pkgbase + \
                    '@aur.archlinux.org>'
        return headers_reply(thread_id)


class FlagNotification(Notification):
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
    def __init__(self, uid, pkgbase_id):

        self._user = db.query(User.Username).filter(
            User.ID == uid).first().Username
        self._pkgbase = db.query(PackageBase.Name).filter(
            PackageBase.ID == pkgbase_id).first().Name

        query = db.query(User).join(PackageComaintainer, isouter=True).join(
            PackageBase,
            or_(PackageBase.MaintainerUID == User.ID,
                PackageBase.ID == PackageComaintainer.PackageBaseID)
        ).filter(
            and_(PackageBase.ID == pkgbase_id,
                 User.Suspended == 0)
        ).with_entities(
            User.Email,
            User.LangPreference
        ).distinct()
        self._recipients = [(u.Email, u.LangPreference) for u in query]

        pkgbase = db.query(PackageBase.FlaggerComment).filter(
            PackageBase.ID == pkgbase_id).first()
        self._text = pkgbase.FlaggerComment

Lukas Fleischer's avatar
Lukas Fleischer committed
320
        super().__init__()
321
322

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
323
        return self._recipients
324

Lukas Fleischer's avatar
Lukas Fleischer committed
325
    def get_subject(self, lang):
326
327
328
        return aurweb.l10n.translator.translate(
            'AUR Out-of-date Notification for {pkgbase}',
            lang).format(pkgbase=self._pkgbase)
329

Lukas Fleischer's avatar
Lukas Fleischer committed
330
    def get_body(self, lang):
331
332
333
334
        body = aurweb.l10n.translator.translate(
            'Your package {pkgbase} [1] has been flagged out-of-date by '
            '{user} [2]:', lang).format(pkgbase=self._pkgbase,
                                        user=self._user)
335
336
337
338
339
340
341
342
343
        body += '\n\n' + self._text
        return body

    def get_refs(self):
        return (aur_location + '/pkgbase/' + self._pkgbase + '/',
                aur_location + '/account/' + self._user + '/')


class OwnershipEventNotification(Notification):
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
    def __init__(self, uid, pkgbase_id):

        self._user = db.query(User.Username).filter(
            User.ID == uid).first().Username
        self._pkgbase = db.query(PackageBase.Name).filter(
            PackageBase.ID == pkgbase_id).first().Name

        query = db.query(User).join(PackageNotification).filter(
            and_(User.OwnershipNotify == 1,
                 PackageNotification.UserID != uid,
                 PackageNotification.PackageBaseID == pkgbase_id,
                 User.Suspended == 0)
        ).with_entities(
            User.Email,
            User.LangPreference
        ).distinct()
        self._recipients = [(u.Email, u.LangPreference) for u in query]

        pkgbase = db.query(PackageBase.FlaggerComment).filter(
            PackageBase.ID == pkgbase_id).first()
        self._text = pkgbase.FlaggerComment

Lukas Fleischer's avatar
Lukas Fleischer committed
366
        super().__init__()
367
368

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
369
        return self._recipients
370

Lukas Fleischer's avatar
Lukas Fleischer committed
371
    def get_subject(self, lang):
372
373
374
        return aurweb.l10n.translator.translate(
            'AUR Ownership Notification for {pkgbase}',
            lang).format(pkgbase=self._pkgbase)
375
376
377
378
379
380
381

    def get_refs(self):
        return (aur_location + '/pkgbase/' + self._pkgbase + '/',
                aur_location + '/account/' + self._user + '/')


class AdoptNotification(OwnershipEventNotification):
Lukas Fleischer's avatar
Lukas Fleischer committed
382
    def get_body(self, lang):
383
384
385
        return aurweb.l10n.translator.translate(
            'The package {pkgbase} [1] was adopted by {user} [2].',
            lang).format(pkgbase=self._pkgbase, user=self._user)
386
387
388


class DisownNotification(OwnershipEventNotification):
389
    def get_body(self, lang):
390
391
392
393
        return aurweb.l10n.translator.translate(
            'The package {pkgbase} [1] was disowned by {user} '
            '[2].', lang).format(pkgbase=self._pkgbase,
                                 user=self._user)
394
395
396


class ComaintainershipEventNotification(Notification):
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
    def __init__(self, uid, pkgbase_id):

        self._pkgbase = db.query(PackageBase.Name).filter(
            PackageBase.ID == pkgbase_id).first().Name

        user = db.query(User).filter(
            and_(User.ID == uid,
                 User.Suspended == 0)
        ).with_entities(
            User.Email,
            User.LangPreference
        ).first()

        self._to = user.Email
        self._lang = user.LangPreference

Lukas Fleischer's avatar
Lukas Fleischer committed
413
        super().__init__()
414
415

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
416
        return [(self._to, self._lang)]
417

Lukas Fleischer's avatar
Lukas Fleischer committed
418
    def get_subject(self, lang):
419
420
421
        return aurweb.l10n.translator.translate(
            'AUR Co-Maintainer Notification for {pkgbase}',
            lang).format(pkgbase=self._pkgbase)
422
423
424
425
426
427

    def get_refs(self):
        return (aur_location + '/pkgbase/' + self._pkgbase + '/',)


class ComaintainerAddNotification(ComaintainershipEventNotification):
Lukas Fleischer's avatar
Lukas Fleischer committed
428
    def get_body(self, lang):
429
430
431
        return aurweb.l10n.translator.translate(
            'You were added to the co-maintainer list of {pkgbase} [1].',
            lang).format(pkgbase=self._pkgbase)
432
433
434


class ComaintainerRemoveNotification(ComaintainershipEventNotification):
Lukas Fleischer's avatar
Lukas Fleischer committed
435
    def get_body(self, lang):
436
437
438
        return aurweb.l10n.translator.translate(
            'You were removed from the co-maintainer list of {pkgbase} '
            '[1].', lang).format(pkgbase=self._pkgbase)
439
440
441


class DeleteNotification(Notification):
442
443
444
445
446
447
448
449
    def __init__(self, uid, old_pkgbase_id, new_pkgbase_id=None):

        self._user = db.query(User.Username).filter(
            User.ID == uid).first().Username
        self._old_pkgbase = db.query(PackageBase.Name).filter(
            PackageBase.ID == old_pkgbase_id).first().Name

        self._new_pkgbase = None
450
        if new_pkgbase_id:
451
452
453
454
455
456
457
458
459
460
461
462
463
            self._new_pkgbase = db.query(PackageBase.Name).filter(
                PackageBase.ID == new_pkgbase_id).first().Name

        query = db.query(User).join(PackageNotification).filter(
            and_(PackageNotification.UserID != uid,
                 PackageNotification.PackageBaseID == old_pkgbase_id,
                 User.Suspended == 0)
        ).with_entities(
            User.Email,
            User.LangPreference
        ).distinct()
        self._recipients = [(u.Email, u.LangPreference) for u in query]

Lukas Fleischer's avatar
Lukas Fleischer committed
464
        super().__init__()
465
466

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
467
        return self._recipients
468

Lukas Fleischer's avatar
Lukas Fleischer committed
469
    def get_subject(self, lang):
470
471
472
        return aurweb.l10n.translator.translate(
            'AUR Package deleted: {pkgbase}',
            lang).format(pkgbase=self._old_pkgbase)
473

Lukas Fleischer's avatar
Lukas Fleischer committed
474
    def get_body(self, lang):
475
        if self._new_pkgbase:
476
477
478
479
            dnlabel = aurweb.l10n.translator.translate(
                'Disable notifications', lang)
            return aurweb.l10n.translator.translate(
                '{user} [1] merged {old} [2] into {new} [3].\n\n'
480
                '--\n'
481
482
483
484
                'If you no longer wish receive notifications about the '
                'new package, please go to [3] and click "{label}".',
                lang).format(user=self._user, old=self._old_pkgbase,
                             new=self._new_pkgbase, label=dnlabel)
485
        else:
486
487
488
489
490
            return aurweb.l10n.translator.translate(
                '{user} [1] deleted {pkgbase} [2].\n\n'
                'You will no longer receive notifications about this '
                'package.', lang).format(user=self._user,
                                         pkgbase=self._old_pkgbase)
491
492
493
494
495
496
497
498
499
500

    def get_refs(self):
        refs = (aur_location + '/account/' + self._user + '/',
                aur_location + '/pkgbase/' + self._old_pkgbase + '/')
        if self._new_pkgbase:
            refs += (aur_location + '/pkgbase/' + self._new_pkgbase + '/',)
        return refs


class RequestOpenNotification(Notification):
501
502
503
504
505
506
507
    def __init__(self, uid, reqid, reqtype, pkgbase_id, merge_into=None):

        self._user = db.query(User.Username).filter(
            User.ID == uid).first().Username
        self._pkgbase = db.query(PackageBase.Name).filter(
            PackageBase.ID == pkgbase_id).first().Name

508
        self._to = aurweb.config.get('options', 'aur_request_ml')
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530

        query = db.query(PackageRequest).join(PackageBase).join(
            PackageComaintainer,
            PackageComaintainer.PackageBaseID == PackageRequest.PackageBaseID,
            isouter=True
        ).join(
            User,
            or_(User.ID == PackageRequest.UsersID,
                User.ID == PackageBase.MaintainerUID,
                User.ID == PackageComaintainer.UsersID)
        ).filter(
            and_(PackageRequest.ID == reqid,
                 User.Suspended == 0)
        ).with_entities(
            User.Email
        ).distinct()
        self._cc = [u.Email for u in query]

        pkgreq = db.query(PackageRequest.Comments).filter(
            PackageRequest.ID == reqid).first()

        self._text = pkgreq.Comments
531
532
533
534
535
        self._reqid = int(reqid)
        self._reqtype = reqtype
        self._merge_into = merge_into

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
536
        return [(self._to, 'en')]
537

538
539
540
    def get_cc(self):
        return self._cc

Lukas Fleischer's avatar
Lukas Fleischer committed
541
    def get_subject(self, lang):
542
543
544
        return '[PRQ#%d] %s Request for %s' % \
               (self._reqid, self._reqtype.title(), self._pkgbase)

Lukas Fleischer's avatar
Lukas Fleischer committed
545
    def get_body(self, lang):
546
547
548
549
550
        if self._merge_into:
            body = '%s [1] filed a request to merge %s [2] into %s [3]:' % \
                   (self._user, self._pkgbase, self._merge_into)
            body += '\n\n' + self._text
        else:
551
552
553
            an = 'an' if self._reqtype[0] in 'aeiou' else 'a'
            body = '%s [1] filed %s %s request for %s [2]:' % \
                   (self._user, an, self._reqtype, self._pkgbase)
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
            body += '\n\n' + self._text
        return body

    def get_refs(self):
        refs = (aur_location + '/account/' + self._user + '/',
                aur_location + '/pkgbase/' + self._pkgbase + '/')
        if self._merge_into:
            refs += (aur_location + '/pkgbase/' + self._merge_into + '/',)
        return refs

    def get_headers(self):
        thread_id = '<pkg-request-' + str(self._reqid) + '@aur.archlinux.org>'
        # Use a deterministic Message-ID for the first email referencing a
        # request.
        headers = headers_msgid(thread_id)
        return headers


class RequestCloseNotification(Notification):
Kevin Morris's avatar
Kevin Morris committed
573

574
575
576
577
    def __init__(self, uid, reqid, reason):
        user = db.query(User.Username).filter(User.ID == uid).first()
        self._user = user.Username if user else None

578
        self._to = aurweb.config.get('options', 'aur_request_ml')
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608

        query = db.query(PackageRequest).join(PackageBase).join(
            PackageComaintainer,
            PackageComaintainer.PackageBaseID == PackageRequest.PackageBaseID,
            isouter=True
        ).join(
            User,
            or_(User.ID == PackageRequest.UsersID,
                User.ID == PackageBase.MaintainerUID,
                User.ID == PackageComaintainer.UsersID)
        ).filter(
            and_(PackageRequest.ID == reqid,
                 User.Suspended == 0)
        ).with_entities(
            User.Email
        ).distinct()
        self._cc = [u.Email for u in query]

        pkgreq = db.query(PackageRequest).join(RequestType).filter(
            PackageRequest.ID == reqid
        ).with_entities(
            PackageRequest.ClosureComment,
            RequestType.Name,
            PackageRequest.PackageBaseName
        ).first()

        self._text = pkgreq.ClosureComment
        self._reqtype = pkgreq.Name
        self._pkgbase = pkgreq.PackageBaseName

609
610
611
612
        self._reqid = int(reqid)
        self._reason = reason

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
613
        return [(self._to, 'en')]
614

615
616
617
    def get_cc(self):
        return self._cc

Lukas Fleischer's avatar
Lukas Fleischer committed
618
    def get_subject(self, lang):
619
620
621
622
        return '[PRQ#%d] %s Request for %s %s' % (self._reqid,
                                                  self._reqtype.title(),
                                                  self._pkgbase,
                                                  self._reason.title())
623

Lukas Fleischer's avatar
Lukas Fleischer committed
624
    def get_body(self, lang):
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
        if self._user:
            body = 'Request #%d has been %s by %s [1]' % \
                   (self._reqid, self._reason, self._user)
        else:
            body = 'Request #%d has been %s automatically by the Arch User ' \
                   'Repository package request system' % \
                   (self._reqid, self._reason)
        if self._text.strip() == '':
            body += '.'
        else:
            body += ':\n\n' + self._text
        return body

    def get_refs(self):
        if self._user:
            return (aur_location + '/account/' + self._user + '/',)
        else:
            return ()

    def get_headers(self):
        thread_id = '<pkg-request-' + str(self._reqid) + '@aur.archlinux.org>'
        headers = headers_reply(thread_id)
        return headers


class TUVoteReminderNotification(Notification):
651
    def __init__(self, vote_id):
652
        self._vote_id = int(vote_id)
653
654
655
656
657
658
659
660
661
662
663

        subquery = db.query(TUVote.UserID).filter(TUVote.VoteID == vote_id)
        query = db.query(User).filter(
            and_(User.AccountTypeID.in_((2, 4)),
                 ~User.ID.in_(subquery),
                 User.Suspended == 0)
        ).with_entities(
            User.Email, User.LangPreference
        )
        self._recipients = [(u.Email, u.LangPreference) for u in query]

Lukas Fleischer's avatar
Lukas Fleischer committed
664
        super().__init__()
665
666

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
667
        return self._recipients
668

Lukas Fleischer's avatar
Lukas Fleischer committed
669
    def get_subject(self, lang):
670
671
672
        return aurweb.l10n.translator.translate(
            'TU Vote Reminder: Proposal {id}',
            lang).format(id=self._vote_id)
673

Lukas Fleischer's avatar
Lukas Fleischer committed
674
    def get_body(self, lang):
675
676
677
678
        return aurweb.l10n.translator.translate(
            'Please remember to cast your vote on proposal {id} [1]. '
            'The voting period ends in less than 48 hours.',
            lang).format(id=self._vote_id)
679
680
681

    def get_refs(self):
        return (aur_location + '/tu/?id=' + str(self._vote_id),)
682
683


684
def main():
685
    db.get_engine()
686
687
    action = sys.argv[1]
    action_map = {
688
689
690
691
692
693
694
695
696
697
698
699
700
        'send-resetkey': ResetKeyNotification,
        'welcome': WelcomeNotification,
        'comment': CommentNotification,
        'update': UpdateNotification,
        'flag': FlagNotification,
        'adopt': AdoptNotification,
        'disown': DisownNotification,
        'comaintainer-add': ComaintainerAddNotification,
        'comaintainer-remove': ComaintainerRemoveNotification,
        'delete': DeleteNotification,
        'request-open': RequestOpenNotification,
        'request-close': RequestCloseNotification,
        'tu-vote-reminder': TUVoteReminderNotification,
701
702
    }

703
704
    with db.begin():
        notification = action_map[action](*sys.argv[2:])
705
    notification.send()
706

707
708
709

if __name__ == '__main__':
    main()