def check_write_enabler(storage_index, write_enabler):
db_connection = get_db_connection()
cursor = db_connection.cursor()
cursor.execute('SELECT * FROM write_enablers WHERE storage_index = %s',
(storage_index,))
result = cursor.fetchone()
db_connection.close()
if result is not None:
return result[1] == write_enabler
else:
raise FileSystemException('File does not have storage index registered'
', is it an immutable file?', 403)
return False
def check_write_enabler(storage_index, write_enabler):
db_connection = get_db_connection()
cursor = db_connection.cursor()
cursor.execute('SELECT * FROM write_enablers WHERE storage_index = %s',
(storage_index, ))
result = cursor.fetchone()
db_connection.close()
if result is not None:
return result[1] == write_enabler
else:
raise FileSystemException(
'File does not have storage index registered'
', is it an immutable file?', 403)
return False
def save_file(storage_index, data, data_length, write_enabler=None):
'''Saves a file to the FILE_STORE.
Adds the write enabler to database if given.
Raises FileSystemException if something goes wrong,
returns status code if OK.
The HTTP/1.1 specification [0] dictates return codes.
[0] http://www.w3.org/Protocols/rfc2616/rfc2616-sec9.html
'''
return_code = 500 # Defaults to Internal Server Error
if storage_index is not None:
try:
fd = os.open(os.path.join(FILE_STORE, storage_index),
os.O_WRONLY | os.O_CREAT | os.O_EXCL)
fp = os.fdopen(fd, 'w')
return_code = 201 # Created
if write_enabler is not None:
db_connection = get_db_connection()
cursor = db_connection.cursor()
cursor.execute('INSERT INTO write_enablers VALUES (%s, %s)',
(storage_index, write_enabler))
db_connection.commit()
db_connection.close()
except OSError, e:
if e.errno == errno.EEXIST:
# The file already exist
if check_write_enabler(storage_index, write_enabler):
fp = open(os.path.join(FILE_STORE, storage_index), 'w')
return_code = 200 # OK
else:
# 401 Unauthorized
raise FileSystemException('Wrong write-enabler.', 401)
elif e.errno == errno.EACCES:
# We have problems getting access to FILE_STORE
raise FileSystemException(code=503) # Service Unavailable
else:
# We have some other OSError
raise FileSystemException(errno.errorcode[e.errno], code=500)
#FIXME: Probably want some sort of loop to read chunks at the time.
fp.write(data.read(int(data_length)))
fp.close()
return return_code
def save_file(storage_index, data, data_length, write_enabler=None):
'''Saves a file to the FILE_STORE.
Adds the write enabler to database if given.
Raises FileSystemException if something goes wrong,
returns status code if OK.
The HTTP/1.1 specification [0] dictates return codes.
[0] http://www.w3.org/Protocols/rfc2616/rfc2616-sec9.html
'''
return_code = 500 # Defaults to Internal Server Error
if storage_index is not None:
try:
fd = os.open(os.path.join(FILE_STORE, storage_index),
os.O_WRONLY | os.O_CREAT | os.O_EXCL)
fp = os.fdopen(fd, 'w')
return_code = 201 # Created
if write_enabler is not None:
db_connection = get_db_connection()
cursor = db_connection.cursor()
cursor.execute('INSERT INTO write_enablers VALUES (%s, %s)',
(storage_index, write_enabler))
db_connection.commit()
db_connection.close()
except OSError, e:
if e.errno == errno.EEXIST:
# The file already exist
if check_write_enabler(storage_index, write_enabler):
fp = open(os.path.join(FILE_STORE, storage_index), 'w')
return_code = 200 # OK
else:
# 401 Unauthorized
raise FileSystemException('Wrong write-enabler.', 401)
elif e.errno == errno.EACCES:
# We have problems getting access to FILE_STORE
raise FileSystemException(code=503) # Service Unavailable
else:
# We have some other OSError
raise FileSystemException(errno.errorcode[e.errno], code=500)
#FIXME: Probably want some sort of loop to read chunks at the time.
fp.write(data.read(int(data_length)))
fp.close()
return return_code