ftp_deploy.utils.email: 80 total statements, 93.2% covered

Generated: Sun 2014-03-16 19:26 GMT

Source file: /var/www/service.dev/service/ftp_deploy/utils/email.py

Stats: 68 executed, 5 missed, 7 excluded, 47 ignored

1. import json
3. from abc import ABCMeta, abstractmethod
4. from django.core.mail import EmailMultiAlternatives
5. from django.template.loader import render_to_string
7. from ftp_deploy.conf import *
8. from .repo import commits_parser, repository_parser
9. from .curl import curl_connection
12. class notification():
14. """Notification abstract class, take three arguments, host, service object and payload json string"""
15. __metaclass__ = ABCMeta
17. def __init__(self, host, service, payload):
18. self.host = host
19. self.service = service
20. self.payload = json.loads(payload)
21. self.commits = self.payload['commits']
22. self.from_email = 'noreply@ftpdeploy.com'
23. self.repo = repository_parser(self.payload, self.service)
24. self.send()
26. @property
27. def template_html(self):
28. raise NotImplementedError
30. @property
31. def template_text(self):
32. raise NotImplementedError
34. @abstractmethod
35. def subject(self):
36. pass
38. @abstractmethod
39. def emails(self):
40. pass
42. @abstractmethod
43. def context(self):
44. pass
46. def send(self):
47. """Sent method process emails from list returned by emails() method"""
49. for recipient in self.emails():
50. text_content = render_to_string(self.template_text, self.context())
51. html_content = render_to_string(self.template_html, self.context())
53. msg = EmailMultiAlternatives(self.subject(), text_content, self.from_email, [recipient])
54. msg.attach_alternative(html_content, "text/html")
55. msg.send()
58. class notification_success(notification):
60. """Notification class for success"""
62. template_html = 'ftp_deploy/email/email_success.html'
63. template_text = 'ftp_deploy/email/email_success.txt'
65. def subject(self):
66. return '%s - Deploy Successfully' % self.service
68. def emails(self):
69. emails_list = list()
70. notifications = self.service.notification
72. if notifications:
73. emails_list += notifications.get_success()
75. if notifications.deploy_user_success():
76. emails_list += self.repo.deploy_email()
78. if notifications.commit_user_success():
79. emails_list += commits_parser(self.commits, self.service.repo_source).email_list()
81. return list(set(emails_list))
83. def context(self):
84. context = dict()
85. context['service'] = self.service
86. context['host'] = self.host
87. context['commits_info'] = commits_parser(self.commits, self.service.repo_source).commits_info()
88. context['files_added'], context['files_modified'], context['files_removed'] = commits_parser(self.commits, self.service.repo_source).file_diff()
89. return context
92. class notification_fail(notification):
94. """Notification class for fail"""
96. template_html = 'ftp_deploy/email/email_fail.html'
97. template_text = 'ftp_deploy/email/email_fail.txt'
99. def __init__(self, host, service, payload, error):
100. self.error = error
101. super(notification_fail, self).__init__(host, service, payload)
103. def subject(self):
104. return '%s - Deploy Fail' % self.service
106. def emails(self):
107. emails_list = list()
108. notifications = self.service.notification
110. if notifications:
112. emails_list += notifications.get_fail()
114. if notifications.deploy_user_fail():
115. emails_list += self.repo.deploy_email()
117. if notifications.commit_user_fail():
118. emails_list += commits_parser(self.commits, self.service.repo_source).email_list()
120. return list(set(emails_list))
122. def context(self):
123. context = dict()
124. context['host'] = self.host
125. context['service'] = self.service
126. context['error'] = self.error
127. return context