66 lines
2.0 KiB
Python
66 lines
2.0 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
"""
|
||
|
|
| This file is part of the web2py Web Framework
|
||
|
|
| Copyrighted by Massimo Di Pierro <mdipierro@cs.depaul.edu>
|
||
|
|
| License: LGPLv3 (http://www.gnu.org/licenses/lgpl.html)
|
||
|
|
|
||
|
|
Generates names for cache and session files
|
||
|
|
--------------------------------------------
|
||
|
|
"""
|
||
|
|
import os
|
||
|
|
from gluon._compat import builtin
|
||
|
|
|
||
|
|
def generate(filename, depth=2, base=512):
|
||
|
|
if os.path.sep in filename:
|
||
|
|
path, filename = os.path.split(filename)
|
||
|
|
else:
|
||
|
|
path = None
|
||
|
|
dummyhash = sum(ord(c) * 256 ** (i % 4) for i, c in enumerate(filename)) % base ** depth
|
||
|
|
folders = []
|
||
|
|
for level in range(depth - 1, -1, -1):
|
||
|
|
code, dummyhash = divmod(dummyhash, base ** level)
|
||
|
|
folders.append("%03x" % code)
|
||
|
|
folders.append(filename)
|
||
|
|
if path:
|
||
|
|
folders.insert(0, path)
|
||
|
|
return os.path.join(*folders)
|
||
|
|
|
||
|
|
|
||
|
|
def exists(filename, path=None):
|
||
|
|
if os.path.exists(filename):
|
||
|
|
return True
|
||
|
|
if path is None:
|
||
|
|
path, filename = os.path.split(filename)
|
||
|
|
fullfilename = os.path.join(path, generate(filename))
|
||
|
|
if os.path.exists(fullfilename):
|
||
|
|
return True
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
def remove(filename, path=None):
|
||
|
|
if os.path.exists(filename):
|
||
|
|
return os.unlink(filename)
|
||
|
|
if path is None:
|
||
|
|
path, filename = os.path.split(filename)
|
||
|
|
fullfilename = os.path.join(path, generate(filename))
|
||
|
|
if os.path.exists(fullfilename):
|
||
|
|
return os.unlink(fullfilename)
|
||
|
|
raise IOError
|
||
|
|
|
||
|
|
|
||
|
|
def open(filename, mode="r", path=None):
|
||
|
|
if not path:
|
||
|
|
path, filename = os.path.split(filename)
|
||
|
|
fullfilename = None
|
||
|
|
if not mode.startswith('w'):
|
||
|
|
fullfilename = os.path.join(path, filename)
|
||
|
|
if not os.path.exists(fullfilename):
|
||
|
|
fullfilename = None
|
||
|
|
if not fullfilename:
|
||
|
|
fullfilename = os.path.join(path, generate(filename))
|
||
|
|
if mode.startswith('w') and not os.path.exists(os.path.dirname(fullfilename)):
|
||
|
|
os.makedirs(os.path.dirname(fullfilename))
|
||
|
|
return builtin.open(fullfilename, mode)
|