ftp_deploy.models.service: 67 total statements, 100.0% covered

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

Source file: /var/www/service.dev/service/ftp_deploy/models/service.py

Stats: 57 executed, 0 missed, 10 excluded, 29 ignored

  1. import random
  2. import string
  3. import json
  4. from django.db import models
  5. from django.core.urlresolvers import reverse
  6. from ftp_deploy.utils.core import service_check
  7. from ftp_deploy.conf import *
  8. from .notification import Notification
  9. def repo_choices():
  10. choices = tuple()
  11. if BITBUCKET_SETTINGS['username'] and BITBUCKET_SETTINGS['password']:
  12. choices = choices + (('bb', 'BitBucket'),)
  13. if GITHUB_SETTINGS['username'] and GITHUB_SETTINGS['password']:
  14. choices = choices + (('gh', 'Github'),)
  15. return choices
  16. class Service(models.Model):
  17. ftp_host = models.CharField('Host', max_length=255)
  18. ftp_username = models.CharField('Username', max_length=50)
  19. ftp_password = models.CharField('Password', max_length=50)
  20. ftp_path = models.CharField('Path', max_length=255)
  21. repo_source = models.CharField('Source', max_length=10, choices=repo_choices())
  22. repo_name = models.CharField('Respository Name', max_length=50)
  23. repo_slug_name = models.SlugField('Respository Slug', max_length=50)
  24. repo_branch = models.CharField('Branch', max_length=50)
  25. repo_hook = models.BooleanField(default=False)
  26. secret_key = models.CharField('Secret Key', unique=True, max_length=30, default=lambda: ''.join(random.choice(string.letters + string.digits) for x in range(30)))
  27. status = models.BooleanField(default=True)
  28. status_message = models.TextField()
  29. notification = models.ForeignKey(Notification, null=True, blank=True, on_delete=models.SET_NULL)
  30. created = models.DateTimeField(auto_now_add=True)
  31. updated = models.DateTimeField(auto_now=True)
  32. def __unicode__(self):
  33. return self.repo_name
  34. def deploys(self):
  35. return self.log_set.filter(status=True).count()
  36. def fail_deploys(self):
  37. return self.log_set.filter(status=False).filter(skip=False).count()
  38. def skipped_deploys(self):
  39. return self.log_set.filter(status=False).filter(skip=True).count()
  40. def latest_log_date(self):
  41. return self.log_set.latest('created').created
  42. def latest_log_user(self):
  43. return self.log_set.latest('created').user
  44. def hook_url(self):
  45. return reverse('ftpdeploy_deploy', kwargs={'secret_key': self.secret_key})
  46. def get_logs_tree(self):
  47. """get logs tree for restore deploys. Include all logs since first fail apart of skiped."""
  48. first_fail_log = self.log_set.filter(status=0).order_by('pk')[:1]
  49. logs = self.log_set.filter(skip=0).filter(pk__gte=first_fail_log[0].pk).order_by('pk')
  50. return logs
  51. def lock(self):
  52. return self.task_set.filter(active=True).exists()
  53. def has_queue(self):
  54. return self.task_set.all().exists()
  55. def check(self, **kwargs):
  56. message = list()
  57. fails, message = service_check(self).check_all()
  58. if fails[2]:
  59. self.repo_hook = False
  60. else:
  61. self.repo_hook = True
  62. if True in fails:
  63. self.status_message = '<br>'.join(message)
  64. self.status = False
  65. else:
  66. self.status = True
  67. self.status_message = ''
  68. class Meta:
  69. app_label = 'ftp_deploy'
  70. db_table = 'ftp_deploy_service'