ftp_deploy.utils.ftp: 54 total statements, 100.0% covered

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

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

Stats: 51 executed, 0 missed, 3 excluded, 29 ignored

1. import os
2. from ftplib import FTP
4. from .decorators import check
7. class ftp_connection(object):
8. """FTP Connection helper. Provide methods to manage ftp resources"""
9. def __init__(self, host, username, password, ftp_path):
10. self.host = host
11. self.username = username
12. self.password = password
13. self.ftp_path = self.encode(ftp_path)
14. self.connected = 0
16. def connect(self):
17. """Initialize FTP Connection"""
18. self.ftp = FTP(self.host)
19. self.ftp.login(self.username, self.password)
20. self.connected = 1
22. def create_file(self, file_path, content):
23. """Create file populated with 'content' and save to 'file_path' location"""
24. file_path = self.encode(file_path)
25. self.ftp.storbinary('STOR ' + self.ftp_path + file_path, content)
27. def remove_file(self, file_path):
28. """Remove file from 'file_path' location, and clear empty directories"""
29. file_path = self.encode(file_path)
30. self.ftp.delete(self.ftp_path + file_path)
31. dirname = file_path.split('/')
32. for i in xrange(len(dirname)):
33. current = '/'.join(dirname[:-1 - i])
34. try:
35. self.ftp.rmd(self.ftp_path + current)
36. except Exception, e:
37. return False
39. def make_dirs(self, file_path):
40. """ Create FTP tree directories based on 'file_path'"""
41. file_path = self.encode(file_path)
42. dirname = os.path.dirname(file_path).split('/')
43. for i in xrange(len(dirname)):
44. current = '/'.join(dirname[:i + 1])
45. try:
46. self.ftp.dir(self.ftp_path + current)
47. except Exception, e:
48. self.ftp.mkd(self.ftp_path + current)
50. def encode(self, content):
51. """Encode path string"""
52. return str(content).encode('ascii', 'ignore')
54. def quit(self):
55. """Close FTP Connection"""
56. return self.ftp.quit() if self.connected else False
59. class ftp_check(ftp_connection):
61. """Check FTP Connection, return True if fail"""
63. @check('FTP')
64. def check_ftp_login(self):
65. self.connect()
67. @check('FTP')
68. def check_ftp_path(self):
69. self.ftp.cwd(self.ftp_path)
70. self.ftp.cwd('/')
72. def check_all(self):
74. status = self.check_ftp_login()
75. if status[0] == True:
76. return status
78. status = self.check_ftp_path()
79. if status[0] == True:
80. return status
82. return False, ''