def do_work(ignore_time=False): from purchasing.jobs.job_base import JobStatus turn_off_sqlalchemy_events() jobs = JobStatus.query.filter(JobStatus.status == 'new').all() for job in jobs: task = getattr(nightly_jobs, job.name)(time_override=ignore_time) task.run_job(job) turn_on_sqlalchemy_events() refresh_search_view()
def import_state(filepath): ''' Takes a csv of state contracts and imports them into the DB ''' from purchasing.data.importer.state import main print 'Importing data from {filepath}\n'.format(filepath=filepath) main(filepath) print 'Import finished!' print 'Refreshing search_view...' refresh_search_view() print 'Done!' return
def upload_costars_contract(_file): '''Upload a COSTARS pdf document to S3 Arguments: _file: A werkzeug `FileStorage`_ object Returns: A two-tuple of (the name of the uploaded file, the path/url to the file) ''' filename = secure_filename(_file.filename) if current_app.config['UPLOAD_S3']: try: turn_off_sqlalchemy_events() conn, bucket = connect_to_s3( current_app.config['AWS_ACCESS_KEY_ID'], current_app.config['AWS_SECRET_ACCESS_KEY'], 'costars') file_href = upload_file(filename, bucket, input_file=_file, prefix='/', from_file=True) return filename, file_href except Exception: raise finally: turn_on_sqlalchemy_events() refresh_search_view() else: try: os.mkdir(current_app.config['UPLOAD_DESTINATION']) except: pass filepath = os.path.join(current_app.config['UPLOAD_DESTINATION'], filename) _file.save(filepath) return filename, filepath
def upload_costars_contract(_file): '''Upload a COSTARS pdf document to S3 Arguments: _file: A werkzeug `FileStorage`_ object Returns: A two-tuple of (the name of the uploaded file, the path/url to the file) ''' filename = secure_filename(_file.filename) if current_app.config['UPLOAD_S3']: try: turn_off_sqlalchemy_events() conn, bucket = connect_to_s3( current_app.config['AWS_ACCESS_KEY_ID'], current_app.config['AWS_SECRET_ACCESS_KEY'], 'costars' ) file_href = upload_file(filename, bucket, input_file=_file, prefix='/', from_file=True) return filename, file_href except Exception: raise finally: turn_on_sqlalchemy_events() refresh_search_view() else: try: os.mkdir(current_app.config['UPLOAD_DESTINATION']) except: pass filepath = os.path.join(current_app.config['UPLOAD_DESTINATION'], filename) _file.save(filepath) return filename, filepath