load-kca.py

#!/usr/bin/env python3

import argparse
import os
import sys
from datetime import datetime, timezone

import django

sys.path.append('../cadasta')


os.environ.setdefault("DJANGO_SETTINGS_MODULE", "config.settings.dev")


def load_kca():
    """Load Kivulini data."""
    from organization.models import Organization, Project
    from party.models import Party, TenureRelationship, TenureRelationshipType
    from party.models import TenureRelationshipType
    from spatial.models import SpatialUnit

    org = Organization.objects.create(
        name='KCA', slug='kca',
        description='',
        urls=[''],
        logo='https://s3.amazonaws.com/cadasta-dev-tmp/etl-test/kca.png',
        contacts=[]
    )

    proj = Project.objects.create(
        organization=org,
        name='KCA Pilot Training',
        slug='kca-pilot-training',
        description="""As part of the World Bank Real Estate and Cadastre Project, the KCA is testing various approaches for documenting property rights in Krusha e Madhe. This includes using drone imagery, mobile devices for data collection in the field, and the use of tools such as FieldPapers. This project for the KCA will be for testing the data collection devices.""",
        country='KV',
        extent=('SRID=4326;'
                """POLYGON ((19.973144531249996 41.832735062152615, 19.973144531249996 43.265206318396025, 21.8902587890625 43.265206318396025, 21.8902587890625 41.832735062152615, 19.973144531249996 41.832735062152615))""")
    )
    su1 = SpatialUnit.objects.create(
        geometry=('SRID=4326;POINT(20.631539 42.32537)'),
        project=proj,
        type='BU')
    su2 = SpatialUnit.objects.create(
        geometry=('SRID=4326;POINT(20.636637 42.320297)'),
        project=proj,
        type='PA')
    su3 = SpatialUnit.objects.create(
        geometry=('SRID=4326;POINT(20.636772 42.320147)'),
        project=proj,
        type='PA')
    su4 = SpatialUnit.objects.create(
        geometry=('SRID=4326;POINT(21.16524 42.661555)'),
        project=proj,
        type='PA')

    p1 = Party.objects.create(
        name="Rexhep Dellova",
        type="IN",
        project=proj
    )
    p2 = Party.objects.create(
        name="Nexhmedin Dellova",
        type="IN",
        project=proj
    )
    p3 = Party.objects.create(
        name="Shtepia muze e Ukshin Hotit",
        type="IN",
        project=proj
    )
    jt = TenureRelationshipType.objects.get(id='JT')
    tc = TenureRelationshipType.objects.get(id='TC')
    es = TenureRelationshipType.objects.get(id='ES')
    tr1 = TenureRelationship.objects.create(
        project=proj,
        party=p1,
        spatial_unit=su3,
        tenure_type=tc
    )
    tr2 = TenureRelationship.objects.create(
        project=proj,
        party=p2,
        spatial_unit=su2,
        tenure_type=tc
    )
    tr3 = TenureRelationship.objects.create(
        project=proj,
        party=p3,
        spatial_unit=su1,
        tenure_type=jt
    )


def load_users():
    from organization.models import Organization, OrganizationRole
    from accounts.models import User

    users = [{'username': 'lmeco',
              'full_name': 'Lorena Meco', 'admin': False},
             {'username': 'kahmetaj', 'full_name':
              'Korab Ahmetaj', 'admin': True},
             {'username': 'kcaadmin', 'full_name':
              'KCA Admin', 'admin': True}]

    org = Organization.objects.get(slug='kca')

    for user in users:
        u = User.objects.create(
            username=user['username'],
            email='',
            full_name=user['full_name'],
            email_verified=True,
            last_login=datetime.now(tz=timezone.utc)
        )
        u.set_password('password')
        u.save()
        OrganizationRole.objects.create(
            organization=org, user=u, admin=user['admin']
        )

def drop_users():
    from organization.models import OrganizationRole
    from accounts.models import User
    users = ['kcaadmin', 'lmeco', 'kahmetaj']
    for user in users:
        try:
            u = User.objects.get(username=user)
            o = OrganizationRole.objects.get(user=u)
            o.delete()
            u.delete()
        except:
            pass


def drop_kca():
    from organization.models import Organization
    drop_users()
    try:
        Organization.objects.get(slug='kca').delete()
    except:
        pass


if __name__ == '__main__':
    django.setup()
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--delete", help='Delete all data for this organization', action="store_true")
    args = parser.parse_args()
    if args.delete:
        drop_kca()
    else:
        drop_kca()
        load_kca()
        load_users()