def process_and_create_file(project_id, name, length, mime_type):
"""Base function that handles the hashing of a file name and the creation of
a record in the files collection. This function is use in the nodes module
on assets_create.
:type project_id: string
:param bucket_name: The project id, uset to fetch the gcs bucket.
:type name: string
:param subdir: The the filename (currently used to build the path).
:type length: int
:param subdir: Filesize in bit (in case we start the upload from the js
interface, we get the size for free, otherwise at the moment we
hardcode it to 0)
:type mime_type: string
:param subdir: MIME type used do display/preview the file accordingly
"""
root, ext = os.path.splitext(name)
# Hash name based on file name, user id and current timestamp
hash_name = name + str(current_user.objectid) + str(round(time.time()))
link = hashlib.sha1(hash_name).hexdigest()
link = os.path.join(link[:2], link + ext)
src_dir_path = os.path.join(app.config['UPLOAD_DIR'],
str(current_user.objectid))
# Move the file in designated location
destination_dir = os.path.join(app.config['SHARED_DIR'], link[:2])
if not os.path.isdir(destination_dir):
os.makedirs(destination_dir)
# (TODO) Check if filename already exsits
src_file_path = os.path.join(src_dir_path, name)
dst_file_path = os.path.join(destination_dir, link[3:])
# (TODO) Thread this operation
shutil.move(src_file_path, dst_file_path)
api = system_util.pillar_api()
file_item = File({
'name': link[3:],
'filename': name,
'user': current_user.objectid,
'backend': 'gcs',
'md5': '',
'content_type': mime_type,
'length': length,
'project': project_id
})
file_item.create(api=api)
return file_item
def process_and_create_file(project_id, name, length, mime_type):
"""Base function that handles the hashing of a file name and the creation of
a record in the files collection. This function is use in the nodes module
on assets_create.
:type project_id: string
:param bucket_name: The project id, uset to fetch the gcs bucket.
:type name: string
:param subdir: The the filename (currently used to build the path).
:type length: int
:param subdir: Filesize in bit (in case we start the upload from the js
interface, we get the size for free, otherwise at the moment we
hardcode it to 0)
:type mime_type: string
:param subdir: MIME type used do display/preview the file accordingly
"""
root, ext = os.path.splitext(name)
# Hash name based on file name, user id and current timestamp
hash_name = name + str(current_user.objectid) + str(round(time.time()))
link = hashlib.sha1(hash_name).hexdigest()
link = os.path.join(link[:2], link + ext)
src_dir_path = os.path.join(app.config['UPLOAD_DIR'], str(current_user.objectid))
# Move the file in designated location
destination_dir = os.path.join(app.config['SHARED_DIR'], link[:2])
if not os.path.isdir(destination_dir):
os.makedirs(destination_dir)
# (TODO) Check if filename already exsits
src_file_path = os.path.join(src_dir_path, name)
dst_file_path = os.path.join(destination_dir, link[3:])
# (TODO) Thread this operation
shutil.move(src_file_path, dst_file_path)
api = system_util.pillar_api()
file_item = File({
'name': link[3:],
'filename': name,
'user': current_user.objectid,
'backend': 'gcs',
'md5': '',
'content_type': mime_type,
'length': length,
'project': project_id
})
file_item.create(api=api)
return file_item
def create():
name = request.form['name']
size = request.form['size']
content_type = request.form['type']
root, ext = os.path.splitext(name)
# Hash name based on file name, user id and current timestamp
hash_name = name + str(current_user.objectid) + str(round(time.time()))
link = hashlib.sha1(hash_name).hexdigest()
link = os.path.join(link[:2], link + ext)
src_dir_path = os.path.join(app.config['UPLOAD_DIR'],
str(current_user.objectid))
# Move the file in designated location
destination_dir = os.path.join(app.config['SHARED_DIR'], link[:2])
if not os.path.isdir(destination_dir):
os.makedirs(destination_dir)
# (TODO) Check if filename already exsits
src_file_path = os.path.join(src_dir_path, name)
dst_file_path = os.path.join(destination_dir, link[3:])
# (TODO) Thread this operation
shutil.copy(src_file_path, dst_file_path)
api = SystemUtility.attract_api()
node_file = File({
'name': link,
'filename': name,
'user': current_user.objectid,
'backend': 'cdnsun',
'md5': '',
'content_type': content_type,
'length': size
})
node_file.create(api=api)
thumbnail = node_file.thumbnail_file('s', api=api)
return jsonify(status='success',
data=dict(id=node_file._id, link=thumbnail.link))
def create():
name = request.form['name']
size = request.form['size']
content_type = request.form['type']
root, ext = os.path.splitext(name)
# Hash name based on file name, user id and current timestamp
hash_name = name + str(current_user.objectid) + str(round(time.time()))
link = hashlib.sha1(hash_name).hexdigest()
link = os.path.join(link[:2], link + ext)
src_dir_path = os.path.join(app.config['UPLOAD_DIR'], str(current_user.objectid))
# Move the file in designated location
destination_dir = os.path.join(app.config['SHARED_DIR'], link[:2])
if not os.path.isdir(destination_dir):
os.makedirs(destination_dir)
# (TODO) Check if filename already exsits
src_file_path = os.path.join(src_dir_path, name)
dst_file_path = os.path.join(destination_dir, link[3:])
# (TODO) Thread this operation
shutil.copy(src_file_path, dst_file_path)
api = SystemUtility.attract_api()
node_file = File({
'name': link,
'filename': name,
'user': current_user.objectid,
'backend': 'cdnsun',
'md5': '',
'content_type': content_type,
'length': size
})
node_file.create(api=api)
thumbnail = node_file.thumbnail_file('s', api=api)
return jsonify(status='success', data=dict(id=node_file._id,
link=thumbnail.link))
def assets_create():
name = request.form['name']
parent_id = request.form.get('parent_id')
# Detect filetype by extension (improve by detectin real file type)
root, ext = os.path.splitext(name)
if ext in ['.jpg', '.jpeg', '.png', '.tif', '.tiff']:
filetype = 'image'
elif ext in ['.blend', '.txt', '.zip']:
filetype = 'file'
elif ext in ['.mov', '.avi', '.mp4', '.m4v']:
filetype = 'video'
else:
filetype = 'file'
# Hash name based on file name, user id and current timestamp
hash_name = name + str(current_user.objectid) + str(round(time.time()))
link = hashlib.sha1(hash_name).hexdigest()
link = os.path.join(link[:2], link + ext)
api = SystemUtility.attract_api()
node_type = NodeType.find_first({
'where': '{"name" : "asset"}',
}, api=api)
# We will create the Node object later on, after creating the file object
node_asset_props = dict(
name=name,
#description=a.description,
#picture=picture,
user=current_user.objectid,
node_type=node_type._id,
#parent=node_parent,
properties=dict(
content_type=filetype,
#file=a.link[4:],
status='processing'))
src_dir_path = os.path.join(app.config['UPLOAD_DIR'], str(current_user.objectid))
# Move the file in designated location
destination_dir = os.path.join(app.config['SHARED_DIR'], link[:2])
if not os.path.isdir(destination_dir):
os.makedirs(destination_dir)
# (TODO) Check if filename already exsits
src_file_path = os.path.join(src_dir_path, name)
dst_file_path = os.path.join(destination_dir, link[3:])
# (TODO) Thread this operation
shutil.copy(src_file_path, dst_file_path)
if filetype == 'file':
mime_type = 'application'
else:
mime_type = filetype
content_type = "{0}/{1}".format(mime_type, ext.replace(".", ""))
node_file = File({
'name': link,
'filename': name,
'user': current_user.objectid,
'backend': 'cdnsun',
'md5': '',
'content_type': content_type,
'length': 0
})
node_file.create(api=api)
node_asset_props['properties']['file'] = node_file._id
if parent_id:
node_asset_props['parent'] = parent_id
node_asset = Node(node_asset_props)
node_asset.create(api=api)
return jsonify(
link=link,
name=name,
filetype=filetype,
asset_id=node_asset._id)
def assets_create():
name = request.form['name']
parent_id = request.form.get('parent_id')
# Detect filetype by extension (improve by detectin real file type)
root, ext = os.path.splitext(name)
if ext in ['.jpg', '.jpeg', '.png', '.tif', '.tiff']:
filetype = 'image'
elif ext in ['.blend', '.txt', '.zip']:
filetype = 'file'
elif ext in ['.mov', '.avi', '.mp4', '.m4v']:
filetype = 'video'
else:
filetype = 'file'
# Hash name based on file name, user id and current timestamp
hash_name = name + str(current_user.objectid) + str(round(time.time()))
link = hashlib.sha1(hash_name).hexdigest()
link = os.path.join(link[:2], link + ext)
api = SystemUtility.attract_api()
node_type = NodeType.find_first({
'where': '{"name" : "asset"}',
}, api=api)
# We will create the Node object later on, after creating the file object
node_asset_props = dict(
name=name,
#description=a.description,
#picture=picture,
user=current_user.objectid,
node_type=node_type._id,
#parent=node_parent,
properties=dict(
content_type=filetype,
#file=a.link[4:],
status='processing'))
src_dir_path = os.path.join(app.config['UPLOAD_DIR'],
str(current_user.objectid))
# Move the file in designated location
destination_dir = os.path.join(app.config['SHARED_DIR'], link[:2])
if not os.path.isdir(destination_dir):
os.makedirs(destination_dir)
# (TODO) Check if filename already exsits
src_file_path = os.path.join(src_dir_path, name)
dst_file_path = os.path.join(destination_dir, link[3:])
# (TODO) Thread this operation
shutil.copy(src_file_path, dst_file_path)
if filetype == 'file':
mime_type = 'application'
else:
mime_type = filetype
content_type = "{0}/{1}".format(mime_type, ext.replace(".", ""))
node_file = File({
'name': link,
'filename': name,
'user': current_user.objectid,
'backend': 'cdnsun',
'md5': '',
'content_type': content_type,
'length': 0
})
node_file.create(api=api)
node_asset_props['properties']['file'] = node_file._id
if parent_id:
node_asset_props['parent'] = parent_id
node_asset = Node(node_asset_props)
node_asset.create(api=api)
return jsonify(link=link,
name=name,
filetype=filetype,
asset_id=node_asset._id)
