Python获取JE用户头像 代码优化

xiaoxiao2024-06-11  19

发现昨天写的获取用户头像的代码效率有点低,于是今天重新设计了一下。

上代码:

def getLogo(self, path=None, user=None, remote=False, localForld=APP_LOCAL_LOGO_FORLD_PATH): '''获取用户头像(此方法会将头像保存至本地,下次获取将获取本地图像) @param path:用户头像路径(服务器路径) @param user:用户 @param remote: 是否从远程读取 @param localForld: 本地存储路径 默认:config.APP_LOCAL_LOGO_FORLD_PATH @see: config.APP_LOCAL_LOGO_FORLD_PATH @return: 存储在本地的路径(str) ''' if not path and not user: return None if not path and "logo" in user: path = user["logo"] localpath = self.getLocalLogoPath(path, localForld) if os.path.exists(localpath) and not remote: print "logo has exists in local!" return localpath request = Request(JE_URL + path) f = sendRequest(request) return self.saveLogo(path, f["data"], localForld) def saveLogo(self, path, data, localForld=APP_LOCAL_LOGO_FORLD_PATH): '''将头像保存至本地,生成本地序列号(文件名) @param path:服务器上的路径 @param data:从服务器获取到的头像数据 @param localForld:本地文件夹路径 ''' localPath = self.getLocalLogoPath(path, localForld) if not os.path.exists(localPath) and makeDir(os.path.dirname(localPath)): f = open(localPath, "w") try: f.write(data) return localPath finally: if f:f.close() def getLocalLogoPath(self, path, localForld=APP_LOCAL_LOGO_FORLD_PATH): '''获取服务器logo映射到本地logo的路径 @param path:服务器路径 @param localForld:本地存放logo文件夹 ''' if path[0] == '/': path = path[1:] return os.path.join(localForld, path)

 去掉了流水号作为文件名的机制,采用 服务器路径->本地路径 的直接转换,比如服务器路径为  /upload/xxx/xx/22.jpg,则本地路径就是/home/xx/upload/xxx/xx/22.jpg。其中用到的一个额外的函数(makeDir),是自己写的一个递归创建目录的方法。方法代码如下:

def makeDir(path): '''创建目录,如果父目录不存在则递归创建父目录 @param path: 目录全路径 ''' if os.path.exists(path): return True parentDir = os.path.dirname(path)#获取父目录 if not os.path.exists(parentDir): makeDir(parentDir)#如果父目录不存在,就先创建父目录 os.mkdir(path) return True

 这样就节省了生成流水号,保存流水号和计算流水号带来的一些效率上的损失。

-------------------------------

有时候有些事情想得太复杂了。

转载请注明原文地址: https://www.6miu.com/read-5016057.html

最新回复(0)