mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-01 11:29:48 +08:00
61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
from engine.model.group_table import GroupTable
|
|
from engine.model.file_table import FileTable
|
|
from engine.controller.error_code import ErrorCode
|
|
from engine import db
|
|
|
|
class MetaManager(object):
|
|
|
|
@staticmethod
|
|
def Sync():
|
|
db.session.commit()
|
|
|
|
@staticmethod
|
|
def AddGroup(group_name, dimension):
|
|
new_group = GroupTable(group_name, dimension)
|
|
|
|
# add into database
|
|
db.session.add(new_group)
|
|
|
|
return ErrorCode.SUCCESS_CODE, group_name
|
|
|
|
@staticmethod
|
|
def GetGroup(group_name):
|
|
group = GroupTable.query.filter(GroupTable.group_name==group_name).first()
|
|
if group:
|
|
return ErrorCode.SUCCESS_CODE, group
|
|
else:
|
|
return ErrorCode.FAULT_CODE, None
|
|
|
|
@staticmethod
|
|
def GetAllGroup():
|
|
groups = GroupTable.query.all()
|
|
return groups
|
|
|
|
@staticmethod
|
|
def DeleteGroup(group):
|
|
db.session.delete(group)
|
|
|
|
@staticmethod
|
|
def DeleteGroupFiles(group_name):
|
|
records = FileTable.query.filter(FileTable.group_name == group_name).all()
|
|
for record in records:
|
|
# print("record.group_name: ", record.group_name)
|
|
db.session.delete(record)
|
|
|
|
@staticmethod
|
|
def UpdateGroup(group_name, data):
|
|
GroupTable.query.filter(GroupTable.group_name==group_name).update(data)
|
|
|
|
|
|
@staticmethod
|
|
def GetAllRawFiles(group_name):
|
|
FileTable.query.filter(FileTable.group_name == group_name and FileTable.type == 'raw')
|
|
|
|
@staticmethod
|
|
def CreateRawFile(group_name, filename):
|
|
db.session.add(FileTable(group_name, filename, 'raw', 0))
|
|
|
|
@staticmethod
|
|
def UpdateFile(filename, data):
|
|
FileTable.query.filter(FileTable.filename == filename).update(data)
|