ahserver/ahserver/filestorage.py
2020-07-12 16:09:27 +08:00

49 lines
1.1 KiB
Python

# fileUpload.py
import os
import time
import tempfile
import aiofile
from appPublic.folderUtils import _mkdir
from appPublic.jsonConfig import getConfig
class FileStorage:
def __init__(self):
config = getConfig()
self.root = config.filesroot or tempfile.gettempdir()
def realPath(self,path):
if path[0] == '/':
path = path[1:]
p = os.path.join(self.root,path)
return p
def _name2path(self,name):
name = os.path.basename(name)
paths=[191,193,197,199,97]
v = int(time.time()*1000000)
# b = name.encode('utf8') if not isinstance(name,bytes) else name
# v = int.from_bytes(b,byteorder='big',signed=False)
path = os.path.abspath(os.path.join(self.root,
str(v % paths[0]),
str(v % paths[1]),
str(v % paths[2]),
str(v % paths[3]),
str(v % paths[4]),
name))
return path
async def save(self,name,read_data):
p = self._name2path(name)
_mkdir(os.path.dirname(p))
async with aiofile.AIOFile(p,'wb') as f:
while 1:
d = await read_data()
if not d:
break
await f.write(d)
await f.fsync()
return p[len(self.root):]