From: Karl O. Pinc <kop@karlpinc.com>
Date: Sat, 23 Jan 2021 19:04:02 +0000 (-0600)
Subject: Move initial post values into pgwui_upload_core
X-Git-Url: https://papio.biology.duke.edu/gitweb/?a=commitdiff_plain;h=65eb289647c257f333604f0879673fa3b89a712f;p=pgwui_upload_core

Move initial post values into pgwui_upload_core
---

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):
         '''