I am currently refactoring this package. Use release 0.2.2 for a stable and backwards compatible version. Use release 0.3.0 for new projects. Perspectively, I will remove the Django dependency and will make the package usable in different contexts such as PyMongo or SQL.
Django-etl-sync attemps to derive ETL rules from Django model introspection and is able to trace and create deeply nested relationships such as foreign keys and many-to-many relationships. The user can modify this rules by creating their own sub classes and methods. All Reader, Transformer, and Generator classes can be fully replaced by costum classes. Django forms can be used in place of Transformer classes.
The package currently lacks a method of removing records no longer present in upstream data.
The project was originall developed to synchronize an API with upstream data sources for the Berkeley Ecoinformatics Engine, see https://ecoengine.berkeley.edu/.
Features:
- Re-usable Django app that provides classes for light weight ETL in your project. (Will be more independent from the Django framework in the future)
- Geared toward sync'ing with upstream data sources (e.g. for an API) or legacy data (it was originally build for ecoengine.berkeley.edu loading million records from museum collection data with regular small changes).
- Prioritizes data consistency over speed.
- Subclassing allows for replacing of methods with speedier, simplified or more sophisticated versions.
- Supports data persistence, consistency, normalization, and recreation of relationships from flatten files or dumps.
- Derives ETL rules from Django model introspection (the use of other frameworks or database declarations is planned). This rules can be easily modified and overriden.
- Can be easily used within task cues and parallelization frameworks such as Celery, thorough checks of the state of the target avoids race conditions and inconsistencies (at the cost of speed).
- Python 2.7 upwards, Python 3
- Django 1.7 (tested) upwards
- GDAL if OGR readers for geodata are used
The package is in active development toward a new release. For evaluation, contribution, and testing
pip install -e git+ssh://[email protected]/postfalk/django-etl-sync#egg=django-etl-sync
or for production usage
pip install django-etl-sync
Add etl_sync
to INSTALLED_APPS
in settings.py of your Django project.
The module provides two principal ways of usage on either file or record level.
- Use the
Loader
class to specify all ETL operations. If you need to make changes to the data between reading from the file and writing them to the database create a customTransformer
class (see below). - Use the
Generator
class to generate a Django model instance from a dictionary and return the instance. The input dictionary needs to satisfy all constraints of the model as defined byModelField
attributes. If this is not the case an error will be raised. TheLoader
class will catch this error and create a log entry. Apply transformations before passing the dictionary (or object) to the generator class.
The difference to simply creating an instance by calling Model(**dict)
is a thorough check for consistency and the creation of nested relationships.
# data.txt
record name
1 one
2 two
3 three
# main.py
from django.db import models
from etl_sync.loaders import Loader
class TestModel(models.Model):
"""
Example Model.
"""
record = models.CharField(max_length=10)
name = models.CharField(max_length=10, null=True, blank=True)
class YourLoader(Loader):
"""
Add your specific settings here.
"""
model_class = TestModel
if __name__ == '__main__':
loader = YourLoader(data.txt)
res = loader.load()
# main.py
from etl_sync.generators import BaseInstanceGenerator
from <yourproject>.models import TestModel
dic = {'record': 3, 'name': 'three'}
if __name__ == '__main__':
# add additional transformations here
generator = BaseInstanceGenerator(TestModel)
instance = generator.get_instance(dic)
print(instance, generator.res)
Unique fields
Before loading a record it might be necessary to check whether it already exists, whether it needs to be added or updated (persistence). By default the module inspects the target model and uses model fields with the attribute unique=True
or the model Meta class attribute unique_together
as criterions for persistence. The module will check first whether any record with the given combination of values in unique fields already exists and update that record.
Note
Do not use the models internal pk or id field as identifier for your data! Add an extra field containing the identifier from the upstream source, such as record
or remote_id
.
Extra arguments
Another method to add (or overwrite) persistence criterions is to add a list of fields via key word argument. Obviously, this setting will not be able to to violate model constraints. In that case, an IntegrityError will be raised (or logged when used within the Loader class).
generator = InstanceGenerator(
TestModel, persistence = ['record', 'source'])
Subclassing
You can subclass InstanceGenerator and create your own generator class with a specific persistence criterion.
from etl_sync.generators import InstanceGenerator
class MyGenerator(InstanceGenerator):
"""
My generator class with custom persistence criterion.
"""
persistence = ['record', 'source']
etl_persistence
key in data dictionary
The last method is to put an extra key value pair in your data dictionary, e.g. during dictionary transformation.
dic = {'record': 6365,
'name': 'john',
'occupation': 'developer',
'etl_persistence': ['record']}
This approach is particular helpful for nested records that can be used to create relationships. It seems likely that the related model has different persistence criteria than the model currently loaded. In a recursive call, the InstanceGenerator
might not be
directly accessible (see below). E.g.
dic = {'record': 6565,
'name':
'john',
'occupation': {
'name': 'developer',
'paygroup': 'III',
'etl_persistence': ['name', 'paygroup']}}
If the instance generator is called like this and the create_foreignkey
attribute is True
, the foreign key entry for developer with paygroup III will be generated if not already existent.
In addition, the key value pair etl_create: True
can be set on nested records to create (or prevent the creation if set False
) of nested records.
If record creation is disabled and the persistence criterion cannot be met, the record will be rejected and the rejection logged in the logfile when using the Loader
class.
Defining persistence through concise Django model design is the preferred method. However there might be cases where ETL constraints might be stricter than model constraints.
Once the attribute persistence is set on the Generator
class the model field attributes will be ignored as a source for persistence rules. Nevertheless, conflicts with your Django models will throw IntegrityError
or other database errors.
If the Generator
class is called within the Loader
class, Generator errors will be caught and logged to a logfile, by default in the same folder as the source. The loading process will continue. In contrast, if you use the Generator
class in a different context you need to handle errors in your code
By default django-etl-sync uses the Python csv.DictReader
, other reader classes can be used or created if they are similar (duck-typed) to csv.DictReader
.
The package currently contains a reader for OGR readable files.
from etl_sync.loaders import Loader
from etl_sync.readers import OGRReader
class MyLoader(Loader):
reader_class=OGRReader
Transformations remap the dictionary returned from the reader class to Django model attributes. We attempt to map the
dictionary key to the model field with the matching name. The Transformer
classes allows for remapping and validation of incoming records.
Instantiate InstanceGenerator
with a customized Transformer
class:
from etl_sync.loaders import Loader
from etl_sync.transformes import Transformer
class MyTransformer(Transformer):
mappings = {'id': 'record', 'name': 'last_name'}
defaults = {'last_name': 'Doe'}
forms = []
blacklist = {'last_name': ['NA', r'unknown']}
class MyLoader(Loader):
model_class = SomeModel
transformer_class = MyTransformer
loader = MyLoader(myfile.txt)
loader.load()
- The mapping property contains a dictionary in the form
{‘original_fieldname’: ‘new_fieldname’}
which will remap the dictionary. - The defaults property holds a dictionary that gets applied if the value for the dictionary key in question is empty.
- The forms property holds a list of Django forms that get applied to the dictionary. Be careful, unused keys will not be removed. The new
cleaned_data
keys will be added to the dictionary. - And finally the blacklist property holds a list of values for particular keys that will trigger a validation error. The record will be discarded.
Note
These methods will be applied in exactly that order. If the dictionary changes in one of these steps, the next step needs to take these changes into consideration.
In addition to these built-in transformations, there are two additional methods that can be modified for more thorough changes:
class MyTransformer(Transformer):
def transform(self, dic):
"""Make whatever changes needed here."""
return dic
def validate(self, dic):
"""Raise ValidationErrors"""
if last_name == 'Bunny':
raise ValidationError('I do not want to have this record')
Both methods will be applied after the aforementioned built-in methods encouraging a declarative style.
Django form support
A generic Django form class can also be used as Loader.transformer_class
.
Create transformer for related models
Django-etl-sync will create a log file in the same location as the source file. It will contain the list of rejected records.
- Create readers for more source types, especially for comma limited data, and headerless CSV.
- Add data removal, if deleted from source.
- Improve Documentation, create documention on ReadTheDocs.