notify.py 24.3 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
    def __init__(self, uid, pkgbase_id):

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

402
        user = db.query(User).filter(User.ID == uid).with_entities(
403
404
405
406
407
408
409
            User.Email,
            User.LangPreference
        ).first()

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

Lukas Fleischer's avatar
Lukas Fleischer committed
410
        super().__init__()
411
412

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

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

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


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


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


class DeleteNotification(Notification):
439
440
441
442
443
444
445
446
    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
447
        if new_pkgbase_id:
448
449
450
451
452
453
454
455
456
457
458
459
460
            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
461
        super().__init__()
462
463

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
464
        return self._recipients
465

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

Lukas Fleischer's avatar
Lukas Fleischer committed
471
    def get_body(self, lang):
472
        if self._new_pkgbase:
473
474
475
476
            dnlabel = aurweb.l10n.translator.translate(
                'Disable notifications', lang)
            return aurweb.l10n.translator.translate(
                '{user} [1] merged {old} [2] into {new} [3].\n\n'
477
                '--\n'
478
479
480
481
                '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)
482
        else:
483
484
485
486
487
            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)
488
489
490
491
492
493
494
495
496
497

    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):
498
499
500
501
502
503
504
    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

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

        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
528
529
530
531
532
        self._reqid = int(reqid)
        self._reqtype = reqtype
        self._merge_into = merge_into

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

535
536
537
    def get_cc(self):
        return self._cc

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

Lukas Fleischer's avatar
Lukas Fleischer committed
542
    def get_body(self, lang):
543
544
545
546
547
        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:
548
549
550
            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)
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
            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
570

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

575
        self._to = aurweb.config.get('options', 'aur_request_ml')
576
577
578
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

        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

606
607
608
609
        self._reqid = int(reqid)
        self._reason = reason

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

612
613
614
    def get_cc(self):
        return self._cc

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

Lukas Fleischer's avatar
Lukas Fleischer committed
621
    def get_body(self, lang):
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
        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):
648
    def __init__(self, vote_id):
649
        self._vote_id = int(vote_id)
650
651
652
653
654
655
656
657
658
659
660

        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
661
        super().__init__()
662
663

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
664
        return self._recipients
665

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

Lukas Fleischer's avatar
Lukas Fleischer committed
671
    def get_body(self, lang):
672
673
674
675
        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)
676
677
678

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


681
def main():
682
    db.get_engine()
683
684
    action = sys.argv[1]
    action_map = {
685
686
687
688
689
690
691
692
693
694
695
696
697
        '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,
698
699
    }

700
701
    with db.begin():
        notification = action_map[action](*sys.argv[2:])
702
    notification.send()
703

704
705
706

if __name__ == '__main__':
    main()