def appCreateCatalog():
"创建目录"
curTimestamp = unit.getTimeInt()
result = checkApiParam(True, [
{"name": "parentId", "checkfunc": unit.checkParamForInt},
{"name": "name", "checkfunc": lambda v: v if len(v) >= 1 and len(v) < 100 else None},
{"name": "createTime", "checkfunc": unit.checkParamForTimestamp, "default": curTimestamp},
{"name": "lastModifyTime", "checkfunc": unit.checkParamForTimestamp, "default": curTimestamp},
{"name": "memo", "default": None},
{"name": "helpInt", "default": None},
{"name": "helpText", "default": None}])
if not result[kParamForResult]:
return result[kParamForErrorResponse]
loginInfo = result[kParamForLoginInfo]
param = result[kParamForRequestParams]
#查询数据
db = _getDbManager()
parentItem = db.getCatalogByIdAndRootIds(param["parentId"], loginInfo.rootIdsString)
if not parentItem:
return responseHelp.buildErrorResponseData(responseHelp.kCmdUserError_CatalogIdInValid)
#创建路径
strPath = unit.makeUserCreateCatalogPath(parentItem[dataManager.kCatalogFieldPath], param["name"])
if not strPath:
return responseHelp.buildErrorResponseData(responseHelp.kCmdServerError_DbDataError)
param["path"] = strPath
item = db.makeCatalog(param)
return responseHelp.buildSuccessResponseData(dataManager.buildCatalogInfo(item, db))
def appSetHelpInfo():
"设置指定记录的辅助信息"
# 参数判断
result = checkApiParam(True, [
{"name": "type", "checkfunc": unit.checkParamForInt},
{"name": "id", "checkfunc": unit.checkParamForInt, "default": -1},
{"name": "helpInt", "checkfunc": unit.checkParamForInt, "default": None},
{"name": "helpText", "default": None}])
if not result[kParamForResult]:
return result[kParamForErrorResponse]
userLoginInfo = result[kParamForLoginInfo]
param = result[kParamForRequestParams]
nType = param["type"]
nId = userLoginInfo.userId if nType == 0 else param["id"]
if nType not in (0, 1, 2) or nId <= 0:
return responseHelp.buildErrorResponseData(responseHelp.kCmdUserError_Param)
nHelpId = param["helpInt"]
strHelpText = param["helpText"]
# 设置
if nHelpId != None or strHelpText != None:
db = _getDbManager()
db.setHelpInfo(nType, nId, nHelpId, strHelpText, userLoginInfo.rootIdsString)
return responseHelp.buildSuccessResponseData(None)
return responseHelp.buildErrorResponseData(kCmdUserError_Param)
def _getFileInfo(aOtherParams, aValideFileTypes):
"""获取指定文件资源,返回值参考 checkApiParam, 若成功时,返回值result[kParamForRequestParams]会加上
"_x_file": 资源对应的全路径
"_x_fileInfo": 在数据库中对应的Row信息
:aOtherParam: 除了"id"之后的其它信息,必须为一个可变数组[]
:aValideFileTypes: 有效的文件类型,一个list(int),None 表示不判断
:returns: 参考 checkApiParam
"""
param = aOtherParams if aOtherParams else []
param.append({"name": "id", "checkfunc": unit.checkParamForInt})
result = checkApiParam(True, param)
if not result[kParamForResult]:
return result
loginInfo = result[kParamForLoginInfo]
param = result[kParamForRequestParams]
nId = param["id"]
#获取文件信息
db = _getDbManager();
dbFile = db.getFileByIdAndRootIds(nId, loginInfo.rootIdsString)
if dbFile is None:
return False, responseHelp.buildErrorResponseData(responseHelp.kCmdUserError_NotResource), 301
#文件类型判断
if aValideFileTypes:
nType = dbFile[dataManager.kFileFieldType];
if nType not in aValideFileTypes:
return False, responseHelp.buildErrorResponseData(responseHelp.kCmdUserError_ErrorFileTypeForOpt), 416
#获取文件位置
nCatalog = dbFile[dataManager.kFileFieldRealCatalogId]
dbCatalog = db.getCatalogByIdAndRootIds(nCatalog, loginInfo.rootIdsString);
if dbCatalog is None:
appLog.logObject().error("数据库中的数据有误,找不到对应的路径信息=> 文件ID=%d" % nId);
return False, responseHelp.buildErrorResponseData(responseHelp.kCmdServerError_DbDataError), 500
#文件判断
strFileName = os.path.join(dbCatalog[dataManager.kCatalogFieldPath], \
dbFile[dataManager.kFileFieldFileName]);
if not os.path.isfile(strFileName):
appLog.logObject().error("找不到文件=> 文件ID=%d" % nId);
return False, responseHelp.buildErrorResponseData(responseHelp.kCmdUserError_ResourceHasBeenRemove), 500
param["_x_file"] = strFileName
param["_x_fileInfo"] = dbFile
return result
def appUpdateCatalog():
result = checkApiParam(True, [
{"name": "id", "checkfunc": unit.checkParamForInt},
{"name": "parentId", "checkfunc": unit.checkParamForInt, "default": None},
{"name": "name", "checkfunc": lambda v: v if len(v) >= 1 and len(v) < 100 else None,
"default": None},
{"name": "memo", "default": None},
{"name": "helpInt", "default": None},
{"name": "helpText", "default": None}])
if not result[kParamForResult]:
return result[kParamForErrorResponse]
loginInfo = result[kParamForLoginInfo]
param = result[kParamForRequestParams]
nId = param.pop("id")
unit.filterNullValue(param)
bOK = False
if len(param) > 0:
db = _getDbManager()
bOK = db.updateCatalog(nId, param, loginInfo.rootIdsString)
if bOK:
item = db.getCatalogById(nId)
return responseHelp.buildSuccessResponseData(dataManager.buildCatalogInfo(item, db))
return responseHelp.buildErrorResponseData(responseHelp.kCmdUserError_NotModify)
def appGetHelpInfo():
"获取指定记录的辅助信息"
# 参数判断
result = checkApiParam(True, [
{"name": "type", "checkfunc": unit.checkParamForInt},
{"name": "id", "checkfunc": unit.checkParamForInt, "default": -1}])
if not result[kParamForResult]:
return result[kParamForErrorResponse]
userLoginInfo = result[kParamForLoginInfo]
param = result[kParamForRequestParams]
nType = param["type"]
nId = userLoginInfo.userId if nType == 0 else param["id"]
if nType not in (0, 1, 2) or nId <= 0:
return responseHelp.buildErrorResponseData(responseHelp.kCmdUserError_Param)
# 查询并返回数据
db = _getDbManager()
hi = db.getHelpInfo(nType, nId, userLoginInfo.rootIdsString)
nLen = len(hi) if hi else 0
ltData = {"helpInt": hi[0] if nLen > 0 else None,
"helpText": hi[1] if nLen > 1 else None,
"lastModifyTime": hi[2] if nLen > 2 else None}
unit.filterNullValue(ltData)
return responseHelp.buildSuccessResponseData(ltData)
def appGetShareFile(ext):
"根据分享的KEY获取文件内容"
result = checkApiParam(False, ("shareKey",))
if not result[kParamForResult]:
return result[kParamForErrorResponse]
param = result[kParamForRequestParams]
strFileName = cache.getAppFileCache().get(param["shareKey"])
# print(strFileName)
return responseHelp.sendFile(strFileName) if strFileName else \
responseHelp.buildErrorResponseData(responseHelp.kCmdUserError_Param);
def appGetThumb():
"获取文件的缩略图"
result = _getFileInfo(
[{"name": "level", "checkfunc": unit.checkParamForInt, "default": 0}],
(defines.kFileTypeImage, defines.kFileTypeGif, defines.kFileTypeVideo))
if not result[kParamForResult]:
return result[kParamForErrorResponse]
param = result[kParamForRequestParams]
nLevel = param["level"]
strFile = param["_x_file"]
dbFile = param["_x_fileInfo"]
nErrorCode = None
nOrigFileStatus = dbFile[dataManager.kFileFieldStatusForOrigin]
nStatus = dbFile[dataManager.kFileFieldStatusForThumb] if nLevel == 0 else\
dbFile[dataManager.kFileFieldStatusForScreen]
if nStatus == defines.kFileStatusFromLocal:
#本地生成缩略图
if nOrigFileStatus in (defines.kFileStatusFromLocal, defines.kFileStatusFromUploaded):
#根据参数生成缩略图
nId = dbFile[dataManager.kFileFieldId]
nWidth = dbFile[dataManager.kFileFieldWidth]
nHeight = dbFile[dataManager.kFileFieldHeight]
nType = dbFile[dataManager.kFileFieldType]
nCatalogId = dbFile[dataManager.kFileFieldRealCatalogId]
nOrientation = dbFile[dataManager.kFileFieldOrientation]
strFileOut = unit.generateThumbailImage(nCatalogId, nId, strFile, nOrientation, nWidth, nHeight, nType, nLevel)
if not strFileOut or not os.path.isfile(strFileOut):
#无法生成缩略图
print(strFileOut)
nErrorCode = responseHelp.kCmdUserError_BuildThumbFailed
field = "statusForThumb" if nLevel == 0 else "statusForScreen"
_getDbManager().updateFile(nId, {field: defines.kFileStatusBuildError}, None)
else:
nErrorCode = responseHelp.kCmdUserError_WaitUploading
elif nStatus == defines.kFileStatusFromUploaded:
#客户端上传,直接发送
nCatalogId = dbFile[dataManager.kFileFieldRealCatalogId]
nId = dbFile[dataManager.kFileFieldId]
strFileOut = unit.getFileThumbFullFileName(nCatalogId, nId, nLevel)
elif nStatus == defines.kFileStatusBuildError:
#本地生成错误
nErrorCode = responseHelp.kCmdUserError_BuildThumbFailed
elif nStatus == defines.kFileStatusFromUploading:
nErrorCode = responseHelp.kCmdUserError_WaitUploading
#发送数据
if nErrorCode != None:
return responseHelp.buildErrorResponseData(nErrorCode, 404)
return responseHelp.sendFile(strFileOut);
def appDeleteCatalog():
"删除目录"
result = checkApiParam(True, [{"name": "ids", "checkfunc": unit.checkParamForIntList}])
if not result[kParamForResult]:
return result[kParamForErrorResponse]
loginInfo = result[kParamForLoginInfo]
param = result[kParamForRequestParams]
db = _getDbManager()
try:
db.deleteCatalogs(param["ids"], loginInfo.rootIdsString)
return responseHelp.buildSuccessResponseData("OK")
except Exception as e:
return responseHelp.buildErrorResponseData(responseHelp.kCmdServerError_DeleteError)
def _makeUserRootPath(aUserId):
"""确保指定用户存在根目录,
返回: BOOL(成功与否),
成功: rootIdList->()
失败: response
"""
db = _getDbManager()
rootIdList = db.getUserRootCatalogs(aUserId)
if not rootIdList or len(rootIdList) == 0:
#没有根目录: 根据配置生成默认路径,并添加到数据库
#判断目录
strPath = configs.defaultUserPath(aUserId)
if not os.path.isdir(strPath):
logObject().error("user: {} do not set root path".format(aUserId))
return False, responseHelp.buildErrorResponseData(responseHelp.kCmdServerError_NotSetRootPath)
#生成记录
fStat = os.stat(strPath)
nCreateTime = fStat.st_ctime
if nCreateTime > fStat.st_mtime:
nCreateTime = fStat.st_mtime
if nCreateTime > fStat.st_atime:
nCreateTime = fStat.st_atime
ci = db.makeCatalog({"path": strPath, "createTime": int(nCreateTime), "name": "iSyncRoot"})
nRootId = ci[dataManager.kCatalogFieldRootId] if ci else None
if nRootId == None:
logObject().error("can not add root catelog: {} for user: {}".format(strPath, aUserId))
return False, responseHelp.buildErrorResponseData(responseHelp.kCmdServerError_DbDataError)
db.makeUserAssociate(aUserId, nRootId);
rootIdList = (nRootId,);
#成功,写入session信息
key = LoginInfo.MakeObject(aUserId, rootIdList);
session[defines.kSessionUserKey] = key;
return True, None
def appUploadFileInfo():
"上传文件信息"
funcCheckStatus = lambda v: int(v) if int(v) in (defines.kFileStatusFromLocal, defines.kFileStatusFromUploading) else defines.kFileStatusFromLocal
curTimestamp = unit.getTimeInt()
result = checkApiParam(True, (
{"name": "cid", "checkfunc": unit.checkParamForInt},
{"name": "name", "checkfunc": lambda v: v if len(v) > 0 and len(v) <= 100 else None},
{"name": "size", "checkfunc": unit.checkParamForInt},
{"name": "type", "checkfunc": unit.checkParamForFileType},
{"name": "ext", "checkfunc": lambda v: v if len(v) <= 10 else None},
{"name": "statusForThumb", "checkfunc": funcCheckStatus, "default": defines.kFileStatusFromLocal},
{"name": "statusForScreen", "checkfunc": funcCheckStatus, "default": defines.kFileStatusFromLocal},
{"name": "createTime", "checkfunc": unit.checkParamForTimestamp, "default": curTimestamp},
{"name": "importTime", "checkfunc": unit.checkParamForTimestamp, "default": curTimestamp},
{"name": "lastModifyTime", "checkfunc": unit.checkParamForTimestamp, "default": curTimestamp},
{"name": "duration", "checkfunc": lambda v: float(v), "default": None},
{"name": "width", "checkfunc": unit.checkParamForInt, "default": None},
{"name": "height", "checkfunc": unit.checkParamForInt, "default": None},
{"name": "orientation", "checkfunc": unit.checkParamForInt, "default": None},
{"name": "longitude", "checkfunc": unit.checkParamForDouble, "default": None},
{"name": "latitude", "checkfunc": unit.checkParamForDouble, "default": None},
{"name": "memo", "checkfunc": unit.checkParamForLess1024, "default": None},
{"name": "helpInt", "checkfunc": unit.checkParamForInt, "default": None},
{"name": "helpText", "default": None},))
if not result[kParamForResult]:
return result[kParamForErrorResponse]
loginInfo = result[kParamForLoginInfo]
param = result[kParamForRequestParams]
nCatalogId = param.pop("cid")
#目录信息
db = _getDbManager()
catalogRow = db.getCatalogByIdAndRootIds(nCatalogId, loginInfo.rootIdsString)
if not catalogRow:
return responseHelp.buildErrorResponseData(responseHelp.kCmdUserError_Param)
strFileName = unit.buildOriginFileName(catalogRow[dataManager.kCatalogFieldPath], param["ext"])
param["uploadUserId"] = loginInfo.userId
param["fileName"] = strFileName
param["statusForOrigin"] = defines.kFileStatusFromUploading
param["uploadTime"] = unit.getTimeInt()
nNewFileId = db.addFile(catalogRow[dataManager.kCatalogFieldRootId], nCatalogId, param)
fileRow = db.getFileByIdAndRootIds(nNewFileId, None)
funcForIdRelatePath = lambda : db.getCatalogIdRelatePathInfo(nCatalogId)
return responseHelp.buildSuccessResponseData(dataManager.buildFileInfo(fileRow, funcForIdRelatePath))
def appUpdateFile():
"上传文件信息"
funcCheckStatus = lambda v: int(v) if int(v) in (defines.kFileStatusFromLocal, defines.kFileStatusFromUploading) else defines.kFileStatusFromLocal
result = checkApiParam(True, (
{"name": "id", "checkfunc": unit.checkParamForInt},
{"name": "catalogId", "checkfunc": unit.checkParamForInt, "default": None},
{"name": "name", "checkfunc": lambda v: v if len(v) > 0 and len(v) <= 100 else None, "default": None},
{"name": "ext", "checkfunc": lambda v: v if len(v) <= 10 else None, "default": None},
{"name": "size", "checkfunc": unit.checkParamForInt, "default": None},
{"name": "type", "checkfunc": unit.checkParamForFileType, "default": None},
{"name": "statusForThumb", "checkfunc": funcCheckStatus, "default": None},
{"name": "statusForScreen", "checkfunc": funcCheckStatus, "default": None},
{"name": "createTime", "checkfunc": unit.checkParamForTimestamp, "default": None},
{"name": "importTime", "checkfunc": unit.checkParamForTimestamp, "default": None},
{"name": "lastModifyTime", "checkfunc": unit.checkParamForTimestamp, "default": None},
{"name": "duration", "checkfunc": lambda v: float(v), "default": None},
{"name": "width", "checkfunc": unit.checkParamForInt, "default": None},
{"name": "height", "checkfunc": unit.checkParamForInt, "default": None},
{"name": "orientation", "checkfunc": unit.checkParamForInt, "default": None},
{"name": "longitude", "checkfunc": unit.checkParamForDouble, "default": None},
{"name": "latitude", "checkfunc": unit.checkParamForDouble, "default": None},
{"name": "memo", "checkfunc": unit.checkParamForLess1024, "default": None},
{"name": "helpInt", "checkfunc": unit.checkParamForInt, "default": None},
{"name": "helpText", "default": None},))
if not result[kParamForResult]:
return result[kParamForErrorResponse]
loginInfo = result[kParamForLoginInfo]
param = result[kParamForRequestParams]
nFileId = param.pop("id")
#目录信息
unit.filterNullValue(param)
bOK = False
if len(param) > 0:
db = _getDbManager()
bOK = db.updateFile(nFileId, param, loginInfo.rootIdsString)
if bOK:
fileRow = db.getFileByIdAndRootIds(nFileId, loginInfo.rootIdsString)
funcForIdRelatePath = lambda : db.getCatalogIdRelatePathInfo(fileRow[dataManager.kFileFieldRealCatalogId])
fileInfo = dataManager.buildFileInfo(fileRow, funcForIdRelatePath)
return responseHelp.buildSuccessResponseData(fileInfo)
return responseHelp.buildErrorResponseData(responseHelp.kCmdUserError_NotModify)
def appLogin():
"登陆"
result = checkApiParam(False, ["userName", "password"])
if not result[kParamForResult]:
return result[kParamForErrorResponse]
param = result[kParamForRequestParams]
db = _getDbManager()
row = db.getUser(param["userName"], param["password"])
if row is None:
return responseHelp.buildErrorResponseData(responseHelp.kCmdUserError_LoginNamePassword)
#查询根目录信息
nUserId = row[dataManager.kUserFieldId]
strUserName = row[dataManager.kUserFieldName]
bOK, resp = _makeUserRootPath(nUserId)
if not bOK:
return resp
userInfo = dataManager.buildUserInfo(row)
return responseHelp.buildSuccessResponseData(userInfo)
def appUploadFile():
"上传文件"
# print(request.args)
result = responseHelp.checkApiParamWithRequestValues(True, (
{"name": "id", "checkfunc": unit.checkParamForInt},
{"name": "obp", "checkfunc": unit.checkParamForInt, "default": 0},
{"name": "tbp", "checkfunc": unit.checkParamForInt, "default": 0},
{"name": "sbp", "checkfunc": unit.checkParamForInt, "default": 0},
{"name": "cm", "default": None}
), request.args)
if not result[kParamForResult]:
return result[kParamForErrorResponse]
loginInfo = result[kParamForLoginInfo]
param = result[kParamForRequestParams]
nFileId = param["id"]
db = _getDbManager()
fileRow = db.getFileByIdAndRootIds(nFileId, loginInfo.rootIdsString)
if not fileRow:
return responseHelp.buildErrorResponseData(responseHelp.kCmdUserError_Param)
upFileInfos = {}
orgFullFileName = None
upFileNames = {}
def _myUploadFileStreamFactory(aTotalContentLength, aContentType, aFileName, aContentLength):
"生成文件进入写入操作"
strFullFileName = None
nBeginPos = 0
strDbFiledName = None
if aFileName == "origin":
#处理R原始文件
catalogRow = db.getCatalogByIdAndRootIds(fileRow[dataManager.kFileFieldRealCatalogId],
loginInfo.rootIdsString)
if catalogRow:
strFullFileName = os.path.join(catalogRow[dataManager.kCatalogFieldPath],
fileRow[dataManager.kFileFieldFileName])
nBeginPos = param["obp"]
strDbFiledName = "statusForOrigin"
nonlocal orgFullFileName
orgFullFileName = strFullFileName
elif aFileName == "thumb":
#小缩略图
strFullFileName = unit.getFileThumbFullFileName(fileRow[dataManager.kFileFieldRealCatalogId], nFileId, 0)
nBeginPos = param["tbp"]
strDbFiledName = "statusForThumb"
elif aFileName == "screen":
#大缩略图
strFullFileName = unit.getFileThumbFullFileName(fileRow[dataManager.kFileFieldRealCatalogId], nFileId, 0)
nBeginPos = param["sbp"]
strDbFiledName = "statusForScreen"
if not strFullFileName:
raise RuntimeError("filename is error")
# 判断文件的大小,与传入参数进行对比,若传入文件大小过大,则直接报错.否则根据传入参数进行写操作
nCurFileSize = os.stat(strFullFileName).st_size if os.path.isfile(strFullFileName) else 0
if nBeginPos > nCurFileSize:
raise RuntimeError("begin position is error!")
upFileInfos[strDbFiledName] = defines.kFileStatusFromUploaded
upFileNames[aFileName] = strFullFileName
# print("will open file: ", strFullFileName, "; beginPos: ", nBeginPos)
f = open(strFullFileName, "a+b")
f.truncate(nBeginPos)
f.seek(0, os.SEEK_END)
return f
# 开始接收文件内容
try:
stream,form,files = werkzeug.formparser.parse_form_data(request.environ,
stream_factory = _myUploadFileStreamFactory,
silent = False)
except Exception as e:
for key in upFileInfos.keys():
upFileInfos[key] = defines.kFileStatusFromUploading
db.updateFile(nFileId, upFileInfos, loginInfo.rootIdsString)
log.logObject().error(e)
return responseHelp.buildErrorResponseData(responseHelp.kCmdUserError_Param)
# 更新数据库
if orgFullFileName:
upFileInfos["size"] = os.stat(orgFullFileName).st_size
db.updateFile(nFileId, upFileInfos, loginInfo.rootIdsString)
# 生成返回数据
fileRow = db.getFileByIdAndRootIds(nFileId, loginInfo.rootIdsString)
funcForIdRelatePath = lambda : db.getCatalogIdRelatePathInfo(fileRow[dataManager.kFileFieldRealCatalogId])
fileInfo = dataManager.buildFileInfo(fileRow, funcForIdRelatePath)
result = {"fileInfo": fileInfo}
#写入检验值
strCm = param["cm"]
if strCm:
strCm = strCm.lower()
bSha1 = strCm == "sha1"
bMd5 = not bSha1 and strCm == "md5"
if bSha1 or bMd5:
check = {}
for key, filename in upFileNames.items():
check[key] = unit.SHA1FileWithName(filename) if bSha1 else unit.MD5FileWithName(filename)
result["check"] = check
return responseHelp.buildSuccessResponseData(result)