update
This commit is contained in:
199
rog/models.py
199
rog/models.py
@ -1,5 +1,143 @@
|
||||
from typing import ChainMap
|
||||
from django.contrib.gis.db import models
|
||||
from django.utils.translation import gettext_lazy as _
|
||||
from django.contrib.auth.models import User
|
||||
from django.db.models.signals import post_save, post_delete, pre_save
|
||||
from django.dispatch import receiver
|
||||
import geopandas as gpd
|
||||
from sqlalchemy import *
|
||||
from geoalchemy2 import Geometry, WKTElement
|
||||
import os, zipfile, glob
|
||||
import environ
|
||||
from geo.Postgres import Db
|
||||
from sqlalchemy.sql.functions import mode
|
||||
from .mapping import location_mapping
|
||||
from .choices import LAYER_CHOICES
|
||||
from django.contrib.gis.utils import LayerMapping
|
||||
|
||||
env = environ.Env(DEBUG=(bool, False))
|
||||
environ.Env.read_env(env_file=".env")
|
||||
|
||||
db = Db(dbname=env("POSTGRES_DBNAME"), user=env("POSTGRES_USER"), password=env("POSTGRES_PASS"), host="postgres-db", port=env("PG_PORT"))
|
||||
|
||||
|
||||
class RogUser(models.Model):
|
||||
email=models.EmailField(_('Email'))
|
||||
phone=models.CharField(_('Phone Number'), max_length=55)
|
||||
first_name=models.CharField(_('First Name'), max_length=255)
|
||||
middle_name=models.CharField(_('Middle Name'), max_length=255, blank=True, null=True)
|
||||
last_name=models.CharField(_('last_name'), max_length=255)
|
||||
nickname=models.CharField(_('Nickname'), max_length=255, blank=True, null=True)
|
||||
country=models.CharField(_('Country'), max_length=255, default='Japan')
|
||||
language=models.CharField(_('Language'), max_length=255, default='Japanese')
|
||||
prefecture=models.CharField(_('Prefecture'), max_length=255, blank=True, null=True)
|
||||
sex=models.CharField(_('Sex'), max_length=255, default='unknown', blank=True, null=True)
|
||||
birth_year=models.IntegerField(_('Birth year'), blank=True, null=True)
|
||||
family_structure =models.IntegerField(_('Family Structure'), blank=True, null=True)
|
||||
introducer = models.IntegerField(_('Introducerr'), blank=True, null=True, default=0)
|
||||
level= models.IntegerField(_('Level'), blank=True, null=True, default=0)
|
||||
parammeters=models.CharField(_('Parameters'), max_length=512)
|
||||
created_at=models.DateTimeField(auto_now_add=True)
|
||||
last_updated_user=models.ForeignKey(User, related_name="roguser_updated_user", on_delete=models.DO_NOTHING)
|
||||
last_updated_at=models.DateTimeField(auto_now=True)
|
||||
|
||||
|
||||
class SystemSettings(models.Model):
|
||||
setting_name=models.CharField(_('Settings Name'), max_length=255)
|
||||
version=models.CharField(_('Version'), max_length=10, blank=True, null=True)
|
||||
effective_date=models.DateTimeField()
|
||||
end_date=models.DateTimeField()
|
||||
parammeters=models.CharField(_('Parameters'), max_length=512)
|
||||
created_at=models.DateTimeField(auto_now_add=True)
|
||||
last_updated_user=models.ForeignKey(User, related_name="system_setting_updated_user", on_delete=models.DO_NOTHING)
|
||||
last_updated_at=models.DateTimeField(auto_now=True)
|
||||
|
||||
class Location(models.Model):
|
||||
location_id=models.IntegerField(_('Location Name'), blank=True, null=True)
|
||||
location_name=models.CharField(_('Location Name'), max_length=255)
|
||||
category=models.CharField(_('Category'), max_length=255, blank=True, null=True)
|
||||
zip=models.CharField(_('Zip code'), max_length=12, blank=True, null=True)
|
||||
address = models.CharField(_('Address'), max_length=512, blank=True, null=True)
|
||||
prefecture = models.CharField(_('Prefecture'), max_length=255, blank=True, null=True)
|
||||
area= models.CharField(_('Area'), max_length=255, blank=True, null=True)
|
||||
city= models.CharField(_('City'), max_length=255, blank=True, null=True)
|
||||
photos=models.JSONField(_('Phptos'), blank=True, null=True)
|
||||
videos=models.JSONField(_('Videos'), blank=True, null=True)
|
||||
web_content=models.JSONField(_('Web Content'), blank=True, null=True)
|
||||
tags=models.CharField(_('Tags'), max_length=512, blank=True, null=True)
|
||||
parammeters=models.CharField(_('Parameters'), max_length=512, blank=True, null=True)
|
||||
created_at=models.DateTimeField(auto_now_add=True)
|
||||
last_updated_user=models.ForeignKey(User, related_name="location_updated_user", on_delete=models.DO_NOTHING,blank=True, null=True)
|
||||
last_updated_at=models.DateTimeField(auto_now=True)
|
||||
geom=models.MultiPointField(srid=4326)
|
||||
|
||||
|
||||
ROG_STATUS = (
|
||||
("REGISTERED", _("Registered")),
|
||||
("ACCEPTED", _("accepted")),
|
||||
("PAID", _("paid")),
|
||||
("JOINED", _("joined")),
|
||||
("CANCELED", _("Canceled"))
|
||||
)
|
||||
|
||||
class JoinedEvent(models.Model):
|
||||
user=models.ForeignKey(User, on_delete=models.DO_NOTHING)
|
||||
tag_name=models.CharField(_('Tag Name'), max_length=255, blank=True, null=True)
|
||||
status=models.CharField(max_length=256, choices=ROG_STATUS)
|
||||
registration_id=models.CharField(_('Registration Id'), max_length=56)
|
||||
payment_code=models.CharField(_('Payment Code'), max_length=255)
|
||||
paied_amount=models.IntegerField(_('Paid Amount'), default=0)
|
||||
remark=models.CharField(_('Remark'), max_length=255, blank=True, null=True)
|
||||
parammeters=models.CharField(_('Parameters'), max_length=512)
|
||||
created_at=models.DateTimeField(auto_now_add=True)
|
||||
last_updated_user=models.ForeignKey(User, related_name="joined_event_updated_user", on_delete=models.DO_NOTHING)
|
||||
last_updated_at=models.DateTimeField(auto_now=True)
|
||||
|
||||
|
||||
class Favorite(models.Model):
|
||||
user=models.ForeignKey(User, on_delete=models.DO_NOTHING)
|
||||
location=models.ForeignKey(Location, on_delete=models.CASCADE)
|
||||
good=models.IntegerField(_('Good'), default=0)
|
||||
favorite=models.IntegerField(_('Favorite'), default=0)
|
||||
evaluation=models.IntegerField(_('Evaluation'), default=0)
|
||||
number_visit=models.IntegerField(_('Good'), default=0)
|
||||
last_visited=models.DateTimeField(_('Last Visited'))
|
||||
parammeters=models.CharField(_('Parameters'), max_length=512)
|
||||
created_at=models.DateTimeField(auto_now_add=True)
|
||||
last_updated_user=models.ForeignKey(User, related_name="favorite_updated_user", on_delete=models.DO_NOTHING)
|
||||
last_updated_at=models.DateTimeField(auto_now=True)
|
||||
|
||||
|
||||
class TravelList(models.Model):
|
||||
travel_id= models.IntegerField(_('Travel Id'))
|
||||
user=models.ForeignKey(User, on_delete=models.DO_NOTHING)
|
||||
start_date=models.DateTimeField()
|
||||
finish_date=models.DateTimeField()
|
||||
category=models.CharField(_('Category'), max_length=255)
|
||||
title=models.CharField(_('Title'), max_length=255)
|
||||
transportation=models.CharField(_('Transpotation'), max_length=255)
|
||||
moving_distance=models.IntegerField()
|
||||
duration=models.DurationField(_('Duration'))
|
||||
eta=models.DateTimeField()
|
||||
parammeters=models.CharField(_('Parameters'), max_length=512)
|
||||
created_at=models.DateTimeField(auto_now_add=True)
|
||||
last_updated_user=models.ForeignKey(User, related_name="travel_list_updated_user", on_delete=models.DO_NOTHING)
|
||||
last_updated_at=models.DateTimeField(auto_now=True)
|
||||
|
||||
|
||||
class TravelPoint(models.Model):
|
||||
travel_id= models.IntegerField(_('Travel Id'))
|
||||
location=models.ForeignKey(Location, on_delete=models.CASCADE)
|
||||
distance=models.FloatField()
|
||||
transportation=models.CharField(_('Transpotation'), max_length=255)
|
||||
eta=models.DateTimeField()
|
||||
order_number=models.IntegerField()
|
||||
parammeters=models.CharField(_('Parameters'), max_length=512)
|
||||
created_at=models.DateTimeField(auto_now_add=True)
|
||||
last_updated_user=models.ForeignKey(User, related_name="travelpoint_updated_user", on_delete=models.DO_NOTHING)
|
||||
last_updated_at=models.DateTimeField(auto_now=True)
|
||||
|
||||
|
||||
|
||||
|
||||
class RogEvent(models.Model):
|
||||
@ -34,4 +172,63 @@ class ShopRoute(models.Model):
|
||||
geom = models.MultiLineStringField(srid=4326)
|
||||
|
||||
def __str__(self):
|
||||
return self.name
|
||||
return self.name
|
||||
|
||||
|
||||
def getTableForModel(tbl):
|
||||
if tbl == 1:
|
||||
return Location.objects.model._meta.db_table;
|
||||
|
||||
def getMappingforModel(tbl, shp):
|
||||
if tbl == 1:
|
||||
return LayerMapping(Location, shp, location_mapping, transform=False)
|
||||
|
||||
|
||||
class ShapeLayers(models.Model):
|
||||
name = models.CharField(_("Shape Layer"), max_length=255)
|
||||
file = models.FileField(upload_to='%y%m%d', blank=True)
|
||||
uploaded_date = models.DateField(auto_now_add=True)
|
||||
layerof = models.IntegerField(choices=LAYER_CHOICES, default=1)
|
||||
table_name = models.CharField(_("Table name"), max_length=255, blank=True)
|
||||
|
||||
def __str__(self):
|
||||
return self.name
|
||||
|
||||
|
||||
|
||||
|
||||
@receiver(pre_save, sender=ShapeLayers)
|
||||
def my_callback(sender, instance, *args, **kwargs):
|
||||
instance.table_name = getTableForModel(instance.layerof)
|
||||
|
||||
@receiver(post_save, sender=ShapeLayers)
|
||||
def publish_date(sender, instance, created, **kwargs):
|
||||
file = instance.file.path
|
||||
file_format = os.path.basename(file).split('.')[-1]
|
||||
file_name = os.path.basename(file).split('.')[0]
|
||||
file_path = os.path.dirname(file)
|
||||
name = instance.name
|
||||
conn_str = f'postgresql://{env("POSTGRES_USER")}:{env("POSTGRES_PASS")}@{env("PG_HOST")}:{env("PG_PORT")}/{env("POSTGRES_DBNAME")}'
|
||||
|
||||
with zipfile.ZipFile(file, 'r') as zip_ref:
|
||||
zip_ref.extractall(file_path)
|
||||
|
||||
os.remove(file)
|
||||
|
||||
try:
|
||||
shp = glob.glob(r'{}/**/*.shp'.format(file_path), recursive=True)[0]
|
||||
gdf = gpd.read_file(shp)
|
||||
crs_name = str(gdf.crs.srs)
|
||||
print(crs_name, 'crs - name')
|
||||
epsg = int(crs_name.replace('epsg:',''))
|
||||
if epsg is None:
|
||||
epsg=4326
|
||||
|
||||
print("### shape file is ###")
|
||||
|
||||
lm2 = getMappingforModel(instance.layerof, shp)
|
||||
lm2.save(strict=True, verbose=True)
|
||||
|
||||
os.remove(shp)
|
||||
except Exception as e:
|
||||
print('##################',e)
|
||||
|
||||
Reference in New Issue
Block a user