ftp_deploy.utils.deploy: 95 total statements, 91.6% covered

Generated: Sat 2014-02-08 14:54 GMT

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

Stats: 76 executed, 7 missed, 12 excluded, 37 ignored

1. import os
2. import json
3. from ftplib import FTP
4. import tempfile
5. import time
6. from celery import task, current_task
8. from ftp_deploy.conf import *
9. from ftp_deploy.models import Log, Task
10. from .core import LockError, commits_parser
11. from .ftp import ftp_connection
12. from .email import notification_success, notification_fail
13. from .curl import curl_connection
16. class Deploy(object):
18. """Deploy method responsible for perform deploying"""
20. def __init__(self, host, payload, service, task_name):
21. self.host = host
22. self.service = service
23. self.task = Task.objects.get(name=task_name)
24. self.data = json.loads(payload)
25. self.json_string = payload
26. self.files_count = commits_parser(self.data['commits']).files_count()
28. self.ftp_host = self.service.ftp_host
29. self.ftp_username = self.service.ftp_username
30. self.ftp_password = self.service.ftp_password
31. self.ftp_path = self.service.ftp_path
33. self.bitbucket_username = BITBUCKET_SETTINGS['username']
34. self.bitbucket_password = BITBUCKET_SETTINGS['password']
36. def perform(self):
37. """Perform ftp connection and choose repository perform method (bitbucket or github)"""
39. if self.data['user'] == 'Restore':
40. self.service.get_logs_tree().delete()
42. self.log = Log()
43. self.log.payload = self.json_string
44. self.log.service = self.service
45. self.log.save()
47. try:
48. self.ftp = ftp_connection(self.ftp_host, self.ftp_username, self.ftp_password, self.ftp_path)
50. if self.service.lock():
51. raise LockError()
53. self.task.active = True
54. self.task.save()
56. self.ftp.connect()
58. except LockError, e:
59. self.set_fail('Service Locked', e)
60. except Exception, e:
61. self.set_fail('FTP Connection', e)
62. else:
63. try:
65. if self.service.repo_source == 'bb':
66. self.perform_bitbucket()
68. if self.service.repo_source == 'gh':
69. self.perform_github()
71. except Exception, e:
72. self.set_fail(self.data['user'], e)
73. else:
74. self.log.user = self.data['user']
75. self.log.status = True
76. self.log.save()
77. notification_success(self.host, self.service, self.json_string)
79. finally:
80. self.ftp.quit()
81. self.service.check()
82. self.service.save()
83. self.task.delete()
85. def perform_bitbucket(self):
86. """perform bitbucket deploy"""
87. curl = curl_connection(self.bitbucket_username, self.bitbucket_password)
88. curl.authenticate()
89. i = 0
90. for commit in self.data['commits']:
91. for files in commit['files']:
92. file_path = files['file']
94. self.update_progress(i,file_path)
95. i += 1
97. if files['type'] == 'removed':
98. self.ftp.remove_file(file_path)
99. else:
100. url = 'https://api.bitbucket.org/1.0/repositories%sraw/%s/%s' % (self.data['repository']['absolute_url'], commit['node'], file_path)
101. url = str(url.encode('utf-8'))
103. value = curl.perform(url)
104. self.create_file(file_path,value)
106. curl.close()
108. def perform_github(self):
109. pass
111. def update_progress(self,i,file_path):
112. progress_percent = int(100 * float(i) / float(self.files_count))
113. current_task.update_state(state='PROGRESS', meta={'status': progress_percent, 'file': os.path.basename(file_path)})
115. def create_file(self,file_path,value):
116. temp_file = tempfile.NamedTemporaryFile(delete=False)
117. temp_file.write(value)
118. temp_file.close()
119. temp_file = open(temp_file.name, 'rb')
121. self.ftp.make_dirs(file_path)
122. self.ftp.create_file(file_path, temp_file)
124. temp_file.close()
125. os.unlink(temp_file.name)
127. def set_fail(self, user, message):
128. self.log.user = user
129. self.log.status_message = message
130. self.log.status = False
131. self.log.save()
132. notification_fail(self.host, self.service, self.json_string, message)