class Server(db.Model, BaseMethod):
'''server list'''
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(32))
ip = db.Column(db.String(16))
port = db.Column(db.Integer)
account = db.Column(db.String(32))
pkey = db.Column(db.Text)
user_id = db.Column(db.String(32), db.ForeignKey(User.id))
user = db.relationship(User)
add_time = db.Column(db.DateTime, default=datetime.datetime.now)
deleted = db.Column(db.Boolean, default=False)
def dict(self, with_pkey=False):
rst = {}
rst['id'] = self.id
rst['name'] = self.name
rst['ip'] = self.ip
rst['port'] = self.port
rst['account'] = self.account
rst['pkey'] = with_pkey and self.pkey or ''
rst['user_id'] = self.user_id
rst['add_time'] = self.add_time
return rst
class History(db.Model, BaseMethod):
'''push history'''
# md5, notice, output, push_name, push_email, success, add_time
id = db.Column(db.Integer, primary_key=True)
status = db.Column(
db.String(1)) # 1:waiting, 2:ing, 3:error, 4:success, 5:except, other
shell_log = db.Column(db.Text) # hook shell log
data = db.Column(db.Text) # git push data
push_user = db.Column(db.String(64)) # git push user(name<email>)
add_time = db.Column(db.DateTime,
default=datetime.datetime.now) # git push time
webhook_id = db.Column(db.Integer, db.ForeignKey(WebHook.id))
webhook = db.relationship(WebHook)
def dict(self):
rst = {}
rst['id'] = self.id
rst['status'] = self.status
rst['shell_log'] = self.shell_log
rst['data'] = self.data # json
rst['push_user'] = self.push_user
rst['add_time'] = self.add_time
rst['webhook_id'] = self.webhook_id
return rst
def updateStatus(self, status):
self.status = status
self.save()
class WebHook(db.Model, BaseMethod):
'''webhook'''
id = db.Column(db.Integer, primary_key=True)
repo = db.Column(db.String(32)) # repo name
branch = db.Column(db.String(32)) # repo branch
shell = db.Column(db.Text) # do what
user_id = db.Column(db.String(32), db.ForeignKey(User.id))
user = db.relationship(User)
server_id = db.Column(db.Integer, db.ForeignKey(Server.id))
server = db.relationship(Server)
add_time = db.Column(db.DateTime, default=datetime.datetime.now)
deleted = db.Column(db.Boolean, default=False)
key = db.Column(db.String(32), unique=True) # 用于webhook，保证私密，直接用 md5 salt
# 1:waiting, 2:ing, 3:error, 4:success, 5:except, other
status = db.Column(db.String(1))
lastUpdate = db.Column(db.DateTime,
default=datetime.datetime.now) # 最新执行时间
def dict(self, with_key=False):
rst = {}
rst['id'] = self.id
rst['repo'] = self.repo
rst['branch'] = self.branch
rst['shell'] = self.shell
rst['user_id'] = self.user_id
rst['server_id'] = self.server_id
rst['server'] = self.server and self.server.dict() or {}
rst['add_time'] = self.add_time
rst['key'] = with_key and self.key or ''
rst['status'] = self.status
rst['lastUpdate'] = self.lastUpdate
return rst
def updateStatus(self, status):
self.status = status
self.lastUpdate = datetime.datetime.now()
self.save()
socketio.emit('webhook',
JsonUtil.object_2_json(self.dict()),
room=self.id)
class Collaborator(db.Model, BaseMethod):
'''Collaborator'''
id = db.Column(db.Integer, primary_key=True)
# webhook
webhook_id = db.Column(db.Integer, db.ForeignKey(WebHook.id))
webhook = db.relationship(WebHook)
# user
user_id = db.Column(db.String(32), db.ForeignKey(User.id))
user = db.relationship(User)
add_time = db.Column(db.DateTime, default=datetime.datetime.now)
def dict(self):
rst = {}
rst['id'] = self.id
rst['webhook_id'] = self.webhook_id
rst['user_id'] = self.user_id
rst['user'] = {}
if self.user:
rst['user'] = self.user.dict()
rst['add_time'] = self.add_time
return rst
class Todo(db.Model, BaseMethod):
'''用户表'''
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
title = db.Column(db.String(1024))
add_time = db.Column(db.DateTime)
completed = db.Column(db.Boolean)
user_id = db.Column(db.String(64), db.ForeignKey('user.id'))
def dict(self):
rst = {}
rst['id'] = self.id
rst['title'] = self.title
rst['add_time'] = self.add_time
rst['completed'] = self.completed
rst['user_id'] = self.user_id
return rst
