Exemple #1
0
def _get_ufile(file_indicator = None):
  ufile = UFile.query.filter(UFile.url == file_indicator).first()
  if not ufile:
    ufile = None
  
  if ufile is not None:
    # test whether file exists
    if not funcs.f_exists(ufile.filename):
      db_session.delete(ufile)
      db_session.commit()
      abort(404)
  else:
    if not funcs.f_exists(file_indicator):
      abort(404)
    
    ufile = UFile(
              name = funcs.get_name_from_filename(file_indicator),
              url = file_indicator,
              filename = file_indicator,
              filesize = funcs.get_file_size(file_indicator),
              mimetype = funcs.get_file_mimetype(file_indicator),
              created = datetime.utcnow()
            )
    db_session.add(ufile)
    db_session.commit()
    #if not r:
    #  logging.error('Failed to commit new record: %r' % ufile)
    #  abort(404)
  return ufile
Exemple #2
0
def gen_file_item(realfile, o_filename):
    pos = o_filename.rfind(".")
    if pos != -1:
        ufile_name = o_filename[:pos]
    else:
        ufile_name = o_filename
    d_fn = os.path.basename(realfile)

    ufile = UFile(
        name=ufile_name,
        url=d_fn,
        filename=d_fn,
        filesize=funcs.get_file_size(d_fn),
        mimetype=funcs.get_file_mimetype(d_fn),
        created=datetime.utcnow(),
    )
    db_session.add(ufile)
    db_session.commit()
    return d_fn