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.
71 lines
2.3 KiB
71 lines
2.3 KiB
#!/usr/bin/env python
|
|
# -*- coding:utf-8 -*-
|
|
# project: 3月
|
|
# author: liuyu
|
|
# date: 2020/3/18
|
|
"""
|
|
本地存储api
|
|
"""
|
|
import logging
|
|
import os
|
|
|
|
from api.utils.TokenManager import make_token
|
|
from api.utils.storage.aliyunApi import AliYunCdn
|
|
from fir_ser import settings
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class LocalStorage(object):
|
|
def __init__(self, domain_name, is_https, download_auth_type=1, cnd_auth_key=None):
|
|
self.domain_name = domain_name
|
|
self.is_https = is_https
|
|
self.download_auth_type = download_auth_type
|
|
self.cnd_auth_key = cnd_auth_key
|
|
|
|
@staticmethod
|
|
def get_upload_token(name, expires):
|
|
return make_token(name, expires)
|
|
|
|
def get_base_url(self):
|
|
uri = 'http://'
|
|
if self.is_https:
|
|
uri = 'https://'
|
|
return f"{uri}{self.domain_name}"
|
|
|
|
def get_download_url(self, name, expires=600, force_new=False):
|
|
download_url = '/'.join([self.get_base_url(), 'download', name])
|
|
if self.download_auth_type == 1:
|
|
download_url = f"{download_url}?{settings.DATA_DOWNLOAD_KEY}={make_token(name, expires, force_new=force_new)}"
|
|
elif self.download_auth_type == 2:
|
|
cdn_obj = AliYunCdn(self.cnd_auth_key, self.is_https, self.domain_name)
|
|
download_url = cdn_obj.get_cdn_download_token(name, expires)
|
|
elif self.download_auth_type == 0:
|
|
pass
|
|
return download_url
|
|
|
|
@staticmethod
|
|
def del_file(name):
|
|
file = os.path.join(settings.MEDIA_ROOT, name)
|
|
try:
|
|
if os.path.isfile(file):
|
|
os.remove(file)
|
|
return True
|
|
except Exception as e:
|
|
logger.error(f"delete file {file} failed Exception {e}")
|
|
return False
|
|
|
|
@staticmethod
|
|
def rename_file(old_filename, new_filename):
|
|
try:
|
|
os.rename(os.path.join(settings.MEDIA_ROOT, old_filename), os.path.join(settings.MEDIA_ROOT, new_filename))
|
|
return True
|
|
except Exception as e:
|
|
logger.error(f"rename_file file {old_filename} to {new_filename} failed Exception {e}")
|
|
return False
|
|
|
|
@staticmethod
|
|
def upload_file(local_file_full_path):
|
|
if os.path.isfile(local_file_full_path):
|
|
return True
|
|
return False
|
|
|