mirror of https://github.com/openspug/spug
72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
# Copyright: (c) OpenSpug Organization. https://github.com/openspug/spug
|
|
# Copyright: (c) <spug.dev@gmail.com>
|
|
# Released under the AGPL-3.0 License.
|
|
from django.db import models
|
|
from django.conf import settings
|
|
from libs.mixins import ModelMixin
|
|
from apps.app.models import App, Environment, Deploy
|
|
from apps.account.models import User
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
import json
|
|
|
|
|
|
class Repository(models.Model, ModelMixin):
|
|
STATUS = (
|
|
('0', '未开始'),
|
|
('1', '构建中'),
|
|
('2', '失败'),
|
|
('5', '成功'),
|
|
)
|
|
app = models.ForeignKey(App, on_delete=models.PROTECT)
|
|
env = models.ForeignKey(Environment, on_delete=models.PROTECT)
|
|
deploy = models.ForeignKey(Deploy, on_delete=models.PROTECT)
|
|
version = models.CharField(max_length=100)
|
|
spug_version = models.CharField(max_length=50)
|
|
remarks = models.CharField(max_length=255, null=True)
|
|
extra = models.TextField()
|
|
status = models.CharField(max_length=2, choices=STATUS, default='0')
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
created_by = models.ForeignKey(User, on_delete=models.PROTECT)
|
|
|
|
@staticmethod
|
|
def make_spug_version(deploy_id):
|
|
return f'{deploy_id}_{datetime.now().strftime("%Y%m%d%H%M%S")}'
|
|
|
|
@property
|
|
def deploy_key(self):
|
|
if self.remarks == 'SPUG AUTO MAKE':
|
|
req = self.deployrequest_set.last()
|
|
if req:
|
|
return f'{settings.REQUEST_KEY}:{req.id}'
|
|
return f'{settings.BUILD_KEY}:{self.id}'
|
|
|
|
def to_view(self):
|
|
tmp = self.to_dict()
|
|
tmp['extra'] = json.loads(self.extra)
|
|
tmp['status_alias'] = self.get_status_display()
|
|
if hasattr(self, 'app_name'):
|
|
tmp['app_name'] = self.app_name
|
|
if hasattr(self, 'env_name'):
|
|
tmp['env_name'] = self.env_name
|
|
if hasattr(self, 'created_by_user'):
|
|
tmp['created_by_user'] = self.created_by_user
|
|
return tmp
|
|
|
|
def delete(self, using=None, keep_parents=False):
|
|
try:
|
|
Path(settings.BUILD_DIR, f'{self.spug_version}.tar.gz').unlink()
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
for p in Path(settings.DEPLOY_DIR).glob(f'{self.deploy_key}:*'):
|
|
try:
|
|
p.unlink()
|
|
except FileNotFoundError:
|
|
pass
|
|
super().delete(using, keep_parents)
|
|
|
|
class Meta:
|
|
db_table = 'repositories'
|
|
ordering = ('-id',)
|