You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.6 KiB
52 lines
1.6 KiB
#!/usr/bin/env python
|
|
# -*- coding:utf-8 -*-
|
|
# project: 3月
|
|
# author: liuyu
|
|
# date: 2020/3/18
|
|
'''
|
|
本地存储api
|
|
'''
|
|
from api.utils.TokenManager import DownloadToken
|
|
from fir_ser import settings
|
|
import os
|
|
|
|
|
|
class LocalStorage(object):
|
|
def __init__(self, domain_name, is_https):
|
|
self.domain_name = domain_name
|
|
self.is_https = is_https
|
|
|
|
def get_upload_token(self, name, expires):
|
|
dtoken = DownloadToken()
|
|
return dtoken.make_token(name, expires)
|
|
|
|
def get_download_url(self, name, expires=600, force_new=False):
|
|
dtoken = DownloadToken()
|
|
base_url = '/'.join([self.domain_name, 'download', name])
|
|
uri = 'http://'
|
|
if self.is_https:
|
|
uri = 'https://'
|
|
if settings.DATA_DOWNLOAD_KEY_OPEN:
|
|
download_url = "%s%s?%s=%s" % (
|
|
uri, base_url, settings.DATA_DOWNLOAD_KEY, dtoken.make_token(name, expires, force_new=force_new))
|
|
else:
|
|
download_url = "%s%s" % (uri, base_url)
|
|
return download_url
|
|
|
|
def del_file(self, name):
|
|
try:
|
|
file = os.path.join(settings.MEDIA_ROOT, name)
|
|
if os.path.isfile(file):
|
|
os.remove(file)
|
|
return True
|
|
except Exception as e:
|
|
print(e)
|
|
return False
|
|
|
|
def rename_file(self, oldfilename, newfilename):
|
|
try:
|
|
os.rename(os.path.join(settings.MEDIA_ROOT, oldfilename), os.path.join(settings.MEDIA_ROOT, newfilename))
|
|
return True
|
|
except Exception as e:
|
|
print(e)
|
|
return False
|
|
|