From 65eb289647c257f333604f0879673fa3b89a712f Mon Sep 17 00:00:00 2001 From: "Karl O. Pinc" Date: Sat, 23 Jan 2021 13:04:02 -0600 Subject: [PATCH] Move initial post values into pgwui_upload_core --- src/pgwui_upload_core/views/upload.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/src/pgwui_upload_core/views/upload.py b/src/pgwui_upload_core/views/upload.py index e03de49..2aa9d36 100644 --- a/src/pgwui_upload_core/views/upload.py +++ b/src/pgwui_upload_core/views/upload.py @@ -31,6 +31,7 @@ import psycopg2.errorcodes from psycopg2 import ProgrammingError from pgwui_core.core import ( + UploadNullFileInitialPost, DataLineProcessor, ParameterExecutor, TabularFileUploadHandler, @@ -44,6 +45,27 @@ from pgwui_upload_core import exceptions as upload_ex log = logging.getLogger(__name__) +@attr.s +class UploadCoreInitialPost(UploadNullFileInitialPost): + '''Get values from settings for when there's not user-supplied + values in POST + ''' + component = attr.ib(default='pgwui_upload') + + def set_component(self, component): + self.component = component + return self + + def build(self, settings={}): + super().build(settings) + upload_settings = settings['pgwui'][self.component] + self.trim_upload = ( + upload_settings['trim'] == 'choice-yes') + self.literal_col_headings = ( + upload_settings['literal_column_headings'] == 'yes') + return self + + class SaveLine(DataLineProcessor, ParameterExecutor): def __init__(self, ue, uh, insert_stmt): ''' -- 2.34.1