add FileUpload object

Author: robswc

src/quickbase_json/helpers.py

@@ -1,3 +1,6 @@
+import base64
+
+
 class IncorrectParameters(Exception):
     def __init__(self, value, expected):
         self.value = value
@@ -93,3 +96,21 @@ def build(self):
             sorters.append({'fieldId': pair[0], 'grouping': pair[1]})
 
         return sorters
+
+
+class FileUpload(dict):
+    """
+    Represents a file object for easy upload to quickbase.
+    When uploading, set FID to FileUpload directly, bypass {'value': etc}
+    """
+    def __init__(self, path: str):
+        """
+        :param path: path to file
+        """
+        super().__init__()
+        self.path = path
+
+        with open(path, 'rb') as f:
+            # get file as a b64 string for upload to quickbase
+            file = base64.b64encode(f.read()).decode()
+            self.update({'value': {'fileName': f'{f.name.split("/")[-1]}', 'data': file}})

src/quickbase_json/qb_response.py

@@ -1,7 +1,7 @@
 import datetime
 
 
-class bcolors:
+class Bcolors:
     HEADER = '\033[95m'
     OKBLUE = '\033[94m'
     OKCYAN = '\033[96m'
@@ -28,13 +28,13 @@ def info(self):
         """
         if self.response_type == 'records':
 
-            print(f'{bcolors.OKBLUE}Sample Data:\n')
+            print(f'{Bcolors.OKBLUE}Sample Data:\n')
             try:
                 print('\t', self.get('data')[0], '\n')
             except KeyError as e:
                 print('\t', self.get('data'), '\n')
-            print(bcolors.ENDC)
-            print(f'{bcolors.OKGREEN}Fields:\n')
+            print(Bcolors.ENDC)
+            print(f'{Bcolors.OKGREEN}Fields:\n')
 
             fields_info = []
             for field in self.get('fields'):
@@ -47,12 +47,12 @@ def info(self):
             # print field info
             for fi in fields_info:
                 print('\t', '{:<16s} {:<32s} {:<16s}'.format(fi[0], fi[1], fi[2]))
-            print(bcolors.ENDC)
+            print(Bcolors.ENDC)
 
-            print(f'\n{bcolors.OKCYAN}Metadata:\n')
+            print(f'\n{Bcolors.OKCYAN}Metadata:\n')
             for k, v in self.get('metadata').items():
                 print('\t{:<16s} {:<16s}'.format(k, str(v)))
-            print(bcolors.ENDC)
+            print(Bcolors.ENDC)
 
     def fields(self, prop):
         """
@@ -63,6 +63,10 @@ def fields(self, prop):
         return [i.get(prop) for i in self.get('fields')]
 
     def data(self):
+        """
+        Gets data, shorthand for .get('data')
+        :return: data
+        """
         return self.get('data')
 
     def prd(self, data):
@@ -204,7 +208,6 @@ def transform(self, transformation, **kwargs):
                                 dt = datetime.datetime.strptime(str_dt, kwargs.get('datestring'))
                                 row.update({str(field.get('id')): {'value': dt}})
 
-
         if transformation == 'intround':
             """Rounds numbers and transforms them to ints"""