def save_file(file_attr, file_data, user): filename = '' try: filename = resource.save(file_data, name=file_attr.name.data + '.') rc = Resource() rc.filename = filename rc.name = file_attr.name.data rc.description = file_attr.description.data rc.user = user rc.level = file_attr.level.data if int(file_attr.level.data) in xrange( 0, 3) else ResourceLevel.PRIVATE rc.usage = file_attr.usage.data if int(file_attr.usage.data) in xrange( 0, 5) else ResourceUsage.OTHER_RES rc.upload_time = datetime.now() rc.type = get_type(rc.file_type) rc.save() return 'OK' except UploadNotAllowed: return 'your upload is not allowed' except IntegrityError: os.remove(resource.path(filename)) db.session.rollback() return 'file name exist' except Exception, e: return 'filed to save you upload'