notify.py 24.2 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
Lukas Fleischer's avatar
Lukas Fleischer committed
14
import aurweb.l10n
15

16
from aurweb import db, logging
17
18
19
from aurweb.models import (PackageBase, PackageComaintainer, PackageComment, PackageNotification, PackageRequest, RequestType,
                           TUVote, User)

20
21
logger = logging.get_logger(__name__)

Lukas Fleischer's avatar
Lukas Fleischer committed
22
aur_location = aurweb.config.get('options', 'aur_location')
23
24


25
26
27
def headers_msgid(thread_id):
    return {'Message-ID': thread_id}

28

29
30
31
def headers_reply(thread_id):
    return {'In-Reply-To': thread_id, 'References': thread_id}

32

33
34
35
36
37
38
39
class Notification:
    def get_refs(self):
        return ()

    def get_headers(self):
        return {}

40
41
42
    def get_cc(self):
        return []

Lukas Fleischer's avatar
Lukas Fleischer committed
43
    def get_body_fmt(self, lang):
44
        body = ''
Lukas Fleischer's avatar
Lukas Fleischer committed
45
        for line in self.get_body(lang).splitlines():
46
47
            if line == '--':
                body += '--\n'
48
                continue
49
50
51
            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
52
        return body.rstrip()
53

54
    def _send(self) -> None:
55
        sendmail = aurweb.config.get('notifications', 'sendmail')
56
57
        sender = aurweb.config.get('notifications', 'sender')
        reply_to = aurweb.config.get('notifications', 'reply-to')
58
59
60
        reason = self.__class__.__name__
        if reason.endswith('Notification'):
            reason = reason[:-len('Notification')]
61
62

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

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

78
79
80
81
82
83
84
85
86
87
            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')
88
89
90
91
92
93
                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')
94
95
96
                user = aurweb.config.get('notifications', 'smtp-user')
                passwd = aurweb.config.get('notifications', 'smtp-password')

97
98
99
100
101
                classes = {
                    False: smtplib.SMTP,
                    True: smtplib.SMTP_SSL,
                }
                server = classes[use_ssl](server_addr, server_port)
102
103
104
105
106
107
108
109

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

                if user and passwd:
                    server.login(user, passwd)
110
111

                server.set_debuglevel(0)
112
                deliver_to = [to] + self.get_cc()
113
                server.sendmail(sender, deliver_to, msg.as_bytes())
114
                server.quit()
115

116
117
118
119
120
121
122
123
    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))

124
125

class ResetKeyNotification(Notification):
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
    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
144
        super().__init__()
145
146

    def get_recipients(self):
147
148
149
150
        if self._backup:
            return [(self._to, self._lang), (self._backup, self._lang)]
        else:
            return [(self._to, self._lang)]
151

Lukas Fleischer's avatar
Lukas Fleischer committed
152
    def get_subject(self, lang):
153
        return aurweb.l10n.translator.translate('AUR Password Reset', lang)
154

Lukas Fleischer's avatar
Lukas Fleischer committed
155
    def get_body(self, lang):
156
157
158
159
160
161
        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)
162
163
164
165
166
167

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


class WelcomeNotification(ResetKeyNotification):
Lukas Fleischer's avatar
Lukas Fleischer committed
168
    def get_subject(self, lang):
169
170
171
        return aurweb.l10n.translator.translate(
            'Welcome to the Arch User Repository',
            lang)
172

Lukas Fleischer's avatar
Lukas Fleischer committed
173
    def get_body(self, lang):
174
175
176
177
178
        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)
179
180
181


class CommentNotification(Notification):
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
    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
204
        super().__init__()
205
206

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
207
        return self._recipients
208

Lukas Fleischer's avatar
Lukas Fleischer committed
209
    def get_subject(self, lang):
210
211
212
        return aurweb.l10n.translator.translate(
            'AUR Comment for {pkgbase}',
            lang).format(pkgbase=self._pkgbase)
213

Lukas Fleischer's avatar
Lukas Fleischer committed
214
    def get_body(self, lang):
215
216
217
        body = aurweb.l10n.translator.translate(
            '{user} [1] added the following comment to {pkgbase} [2]:',
            lang).format(user=self._user, pkgbase=self._pkgbase)
218
        body += '\n\n' + self._text + '\n\n--\n'
219
220
221
222
223
224
        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)
225
226
227
228
229
230
231
232
233
234
235
236
237
        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):
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
    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
256
        super().__init__()
257
258

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
259
        return self._recipients
260

Lukas Fleischer's avatar
Lukas Fleischer committed
261
    def get_subject(self, lang):
262
263
264
        return aurweb.l10n.translator.translate(
            'AUR Package Update: {pkgbase}',
            lang).format(pkgbase=self._pkgbase)
265

Lukas Fleischer's avatar
Lukas Fleischer committed
266
    def get_body(self, lang):
267
268
269
        body = aurweb.l10n.translator.translate(
            '{user} [1] pushed a new commit to {pkgbase} [2].',
            lang).format(user=self._user, pkgbase=self._pkgbase)
270
        body += '\n\n--\n'
271
272
273
274
275
276
        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)
277
278
279
280
281
282
283
284
285
286
287
288
289
        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):
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
    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
314
        super().__init__()
315
316

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
317
        return self._recipients
318

Lukas Fleischer's avatar
Lukas Fleischer committed
319
    def get_subject(self, lang):
320
321
322
        return aurweb.l10n.translator.translate(
            'AUR Out-of-date Notification for {pkgbase}',
            lang).format(pkgbase=self._pkgbase)
323

Lukas Fleischer's avatar
Lukas Fleischer committed
324
    def get_body(self, lang):
325
326
327
328
        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)
329
330
331
332
333
334
335
336
337
        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):
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
    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
360
        super().__init__()
361
362

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
363
        return self._recipients
364

Lukas Fleischer's avatar
Lukas Fleischer committed
365
    def get_subject(self, lang):
366
367
368
        return aurweb.l10n.translator.translate(
            'AUR Ownership Notification for {pkgbase}',
            lang).format(pkgbase=self._pkgbase)
369
370
371
372
373
374
375

    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
376
    def get_body(self, lang):
377
378
379
        return aurweb.l10n.translator.translate(
            'The package {pkgbase} [1] was adopted by {user} [2].',
            lang).format(pkgbase=self._pkgbase, user=self._user)
380
381
382


class DisownNotification(OwnershipEventNotification):
383
    def get_body(self, lang):
384
385
386
387
        return aurweb.l10n.translator.translate(
            'The package {pkgbase} [1] was disowned by {user} '
            '[2].', lang).format(pkgbase=self._pkgbase,
                                 user=self._user)
388
389
390


class ComaintainershipEventNotification(Notification):
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
    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
407
        super().__init__()
408
409

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

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

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


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


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


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

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
461
        return self._recipients
462

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

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

    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):
495
496
497
498
499
500
501
    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

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

        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
525
526
527
528
529
        self._reqid = int(reqid)
        self._reqtype = reqtype
        self._merge_into = merge_into

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

532
533
534
    def get_cc(self):
        return self._cc

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

Lukas Fleischer's avatar
Lukas Fleischer committed
539
    def get_body(self, lang):
540
541
542
543
544
        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:
545
546
547
            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)
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
            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):
567
568
569
570
    def __init__(self, uid, reqid, reason):
        user = db.query(User.Username).filter(User.ID == uid).first()
        self._user = user.Username if user else None

571
        self._to = aurweb.config.get('options', 'aur_request_ml')
572
573
574
575
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

        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

602
603
604
605
        self._reqid = int(reqid)
        self._reason = reason

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

608
609
610
    def get_cc(self):
        return self._cc

Lukas Fleischer's avatar
Lukas Fleischer committed
611
    def get_subject(self, lang):
612
613
614
615
        return '[PRQ#%d] %s Request for %s %s' % (self._reqid,
                                                  self._reqtype.title(),
                                                  self._pkgbase,
                                                  self._reason.title())
616

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

        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
657
        super().__init__()
658
659

    def get_recipients(self):
Lukas Fleischer's avatar
Lukas Fleischer committed
660
        return self._recipients
661

Lukas Fleischer's avatar
Lukas Fleischer committed
662
    def get_subject(self, lang):
663
664
665
        return aurweb.l10n.translator.translate(
            'TU Vote Reminder: Proposal {id}',
            lang).format(id=self._vote_id)
666

Lukas Fleischer's avatar
Lukas Fleischer committed
667
    def get_body(self, lang):
668
669
670
671
        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)
672
673
674

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


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

696
697
    with db.begin():
        notification = action_map[action](*sys.argv[2:])
698
    notification.send()
699

700
701
702

if __name__ == '__main__':
    main()