docassemble.base 1.7.3__py3-none-any.whl

docassemble/base/DA.py

@@ -0,0 +1,138 @@
+__all__ = []
+
+
+class Condition:
+
+    def __init__(self, leftside, operator, rightside):
+        self.leftside = leftside
+        self.operator = operator
+        self.rightside = rightside
+
+    def __and__(self, other):
+        return And(self, other)
+
+    def __or__(self, other):
+        return Or(self, other)
+
+    def __rand__(self, other):
+        return And(other, self)
+
+    def __ror__(self, other):
+        return Or(other, self)
+
+    def __invert__(self):
+        return Condition(self, 'not', None)
+
+    def __repr__(self):
+        return repr(self.leftside) + ' ' + self.operator + ' ' + repr(self.rightside)
+
+
+class Column:
+
+    def __init__(self, name):
+        self.name = name
+
+    def __repr__(self):
+        return self.name
+
+    def __and__(self, other):
+        return Condition(self, 'and', other)
+
+    def __xor__(self, other):
+        return Condition(self, 'xor', other)
+
+    def __or__(self, other):
+        return Condition(self, 'or', other)
+
+    def __rand__(self, other):
+        return Condition(other, 'and', self)
+
+    def __rxor__(self, other):
+        return Condition(other, 'xor', self)
+
+    def __ror__(self, other):
+        return Condition(other, 'or', self)
+
+    def __invert__(self):
+        return Condition(self, 'not', None)
+
+    def __le__(self, other):
+        return Condition(self, 'le', other)
+
+    def __ge__(self, other):
+        return Condition(self, 'ge', other)
+
+    def __gt__(self, other):
+        return Condition(self, 'gt', other)
+
+    def __lt__(self, other):
+        return Condition(self, 'lt', other)
+
+    def __eq__(self, other):
+        return Condition(self, 'eq', other)
+
+    def __ne__(self, other):
+        return Condition(self, 'ne', other)
+
+    def Like(self, rightside):
+        return Condition(self, 'like', rightside)
+
+    def In(self, *pargs):
+        return Condition(self, 'in', list(pargs))
+
+
+class Group:
+
+    def __init__(self, *pargs, **kwargs):  # pylint: disable=unused-argument
+        self.items = list(pargs)
+
+    def __and__(self, other):
+        return And(self, other)
+
+    def __or__(self, other):
+        return Or(self, other)
+
+    def __rand__(self, other):
+        return And(other, self)
+
+    def __ror__(self, other):
+        return Or(other, self)
+
+    def __invert__(self):
+        return Condition(self, 'not', None)
+
+    def __repr__(self):
+        return '(' + self.group_type + ' ' + ', '.join(repr(item) for item in self.items) + ')'
+
+
+class And(Group):
+
+    def __init__(self, *pargs, **kwargs):
+        super().__init__(*pargs, **kwargs)
+        self.group_type = 'and'
+
+
+class Or(Group):
+
+    def __init__(self, *pargs, **kwargs):
+        super().__init__(*pargs, **kwargs)
+        self.group_type = 'or'
+
+
+class Sessions:
+    modtime = Column('modtime')
+    filename = Column('filename')
+    session = Column('key')
+    encrypted = Column('encrypted')
+    user_id = Column('user_id')
+    email = Column('email')
+    first_name = Column('first_name')
+    last_name = Column('last_name')
+    country = Column('country')
+    subdivisionfirst = Column('subdivisionfirst')
+    subdivisionsecond = Column('subdivisionsecond')
+    subdivisionthird = Column('subdivisionthird')
+    organization = Column('organization')
+    timezone = Column('timezone')
+    language = Column('language')
+    last_login = Column('last_login')

docassemble/base/__init__.py

@@ -0,0 +1,5 @@
+try:
+    __import__('pkg_resources').declare_namespace(__name__)
+except ImportError:
+    __path__ = __import__('pkgutil').extend_path(__path__, __name__)
+__version__ = "1.7.3"

docassemble/base/actions.py

@@ -0,0 +1,3 @@
+from docassemble.base.functions import process_action
+
+__all__ = ['process_action']

docassemble/base/amazon.py

@@ -0,0 +1,143 @@
+import os
+import mimetypes
+import datetime
+from botocore.errorfactory import ClientError
+import boto3
+
+epoch = datetime.datetime.utcfromtimestamp(0).replace(tzinfo=datetime.timezone.utc)
+
+
+class s3object:
+
+    def __init__(self, s3_config):
+        self.upload_args = {}
+        self.download_args = {}
+        if 'server side encryption' in s3_config and isinstance(s3_config['server side encryption'], dict):
+            if s3_config['server side encryption'].get('algorithm'):
+                self.upload_args['ServerSideEncryption'] = s3_config['server side encryption']['algorithm']
+            if s3_config['server side encryption'].get('customer algorithm'):
+                self.upload_args['SSECustomerAlgorithm'] = s3_config['server side encryption']['customer algorithm']
+                self.download_args['SSECustomerAlgorithm'] = s3_config['server side encryption']['customer algorithm']
+            if s3_config['server side encryption'].get('customer key'):
+                self.upload_args['SSECustomerKey'] = s3_config['server side encryption']['customer key']
+                self.download_args['SSECustomerKey'] = s3_config['server side encryption']['customer key']
+            if s3_config['server side encryption'].get('KMS key ID'):
+                self.upload_args['SSEKMSKeyId'] = s3_config['server side encryption']['KMS key ID']
+        if 'access key id' in s3_config and s3_config['access key id'] is not None:
+            self.conn = boto3.resource('s3', region_name=s3_config.get('region', None),
+                                       aws_access_key_id=s3_config['access key id'],
+                                       aws_secret_access_key=s3_config['secret access key'],
+                                       endpoint_url=s3_config.get('endpoint url', None))
+            self.client = boto3.client('s3', region_name=s3_config.get('region', None),
+                                       aws_access_key_id=s3_config['access key id'],
+                                       aws_secret_access_key=s3_config['secret access key'],
+                                       endpoint_url=s3_config.get('endpoint url', None))
+        else:
+            self.conn = boto3.resource('s3', region_name=s3_config.get('region', None),
+                                       endpoint_url=s3_config.get('endpoint url', None))
+            self.client = boto3.client('s3', region_name=s3_config.get('region', None),
+                                       endpoint_url=s3_config.get('endpoint url', None))
+        self.bucket = self.conn.Bucket(s3_config['bucket'])
+        self.bucket_name = s3_config['bucket']
+
+    def get_key(self, key_name):
+        return s3key(self, self.conn.Object(self.bucket_name, key_name))
+
+    def search_key(self, key_name):
+        for key in self.bucket.objects.filter(Prefix=key_name, Delimiter='/'):
+            if key.key == key_name:
+                return s3key(self, self.conn.Object(self.bucket_name, key.key))
+        return None
+
+    def list_keys(self, prefix):
+        output = []
+        for obj in self.bucket.objects.filter(Prefix=prefix):
+            new_key = s3key(self, obj)
+            output.append(new_key)
+        return output
+
+
+class s3key:
+
+    def __init__(self, s3_object, key_obj):
+        self.s3_object = s3_object
+        self.key_obj = key_obj
+        self.name = key_obj.key
+        if self.key_obj.__class__.__name__ == 's3.ObjectSummary':
+            self.size = self.key_obj.size
+            self.last_modified = self.key_obj.last_modified
+            self.does_exist = True
+        elif self.exists():
+            self.does_exist = True
+        else:
+            self.does_exist = False
+
+    def get_contents_as_string(self):
+        resp = self.key_obj.get(**self.s3_object.download_args)
+        self.size = resp['ContentLength']
+        self.content_type = resp['ContentType']
+        self.last_modified = resp['LastModified']
+        return resp['Body'].read().decode()
+
+    def exists(self):
+        try:
+            resp = self.s3_object.client.head_object(Bucket=self.s3_object.bucket_name, Key=self.name, **self.s3_object.download_args)
+            self.size = resp['ContentLength']
+            self.content_type = resp['ContentType']
+            self.last_modified = resp['LastModified']
+        except ClientError:
+            return False
+        return True
+
+    def delete(self):
+        self.key_obj.delete()
+        self.does_exist = False
+
+    def get_epoch_modtime(self):
+        return (self.last_modified - epoch).total_seconds()
+
+    def get_contents_to_filename(self, filename):
+        self.s3_object.conn.Bucket(self.s3_object.bucket_name).download_file(self.name, filename, ExtraArgs=self.s3_object.download_args)
+        secs = (self.last_modified - epoch).total_seconds()
+        os.utime(filename, (secs, secs))
+
+    def set_contents_from_filename(self, filename):
+        if hasattr(self, 'content_type') and self.content_type is not None:
+            mimetype = self.content_type
+        else:
+            mimetype, encoding = mimetypes.guess_type(filename)  # pylint: disable=unused-variable
+        if self.key_obj.__class__.__name__ == 's3.ObjectSummary':
+            self.key_obj = self.s3_object.conn.Object(self.s3_object.bucket_name, self.name)
+        if mimetype is not None:
+            self.key_obj.upload_file(filename, ExtraArgs={'ContentType': mimetype, **self.s3_object.upload_args})
+        else:
+            self.key_obj.upload_file(filename, ExtraArgs=self.s3_object.upload_args)
+        resp = self.s3_object.client.head_object(Bucket=self.s3_object.bucket_name, Key=self.name, **self.s3_object.download_args)
+        self.size = resp['ContentLength']
+        self.content_type = resp['ContentType']
+        self.last_modified = resp['LastModified']
+        secs = (self.last_modified - epoch).total_seconds()
+        os.utime(filename, (secs, secs))
+
+    def set_contents_from_string(self, text):
+        if hasattr(self, 'content_type') and self.content_type is not None:
+            self.key_obj.put(Body=bytes(text, encoding='utf-8'), ContentType=self.content_type, **self.s3_object.upload_args)
+        else:
+            self.key_obj.put(Body=bytes(text, encoding='utf-8'), **self.s3_object.upload_args)
+        resp = self.s3_object.client.head_object(Bucket=self.s3_object.bucket_name, Key=self.name, **self.s3_object.download_args)
+        self.size = resp['ContentLength']
+        self.content_type = resp['ContentType']
+        self.last_modified = resp['LastModified']
+
+    def generate_url(self, expires, content_type=None, display_filename=None, inline=False):
+        params = {'Bucket': self.s3_object.bucket_name, 'Key': self.key_obj.key}
+        if content_type is not None:
+            params['ResponseContentType'] = content_type
+        if display_filename is not None:
+            params['ResponseContentDisposition'] = "attachment; filename=" + display_filename
+        if inline:
+            params['ResponseContentDisposition'] = "inline"
+        return self.s3_object.client.generate_presigned_url(
+            ClientMethod='get_object',
+            Params=params,
+            ExpiresIn=expires)