Commit 01c40a7b authored by Vasyl Bodnaruk's avatar Vasyl Bodnaruk

Add export news with tag, technology, industry if exist

parent 5aa75670
import json
from datetime import datetime
from django.db import models
# from esi.webapp.models.wp import WpEsiTagNews
from .wp import WpEsiMedia, WpEsiEntity, WpEsiType, WpEsiRegion, WpEsiNewsEntity, WpEsiNews, WpEsiRadar, WpEsiIndustry, \
WpEsiFunction, WpEsiTechnology, WpEsiTag
WpEsiFunction, WpEsiTechnology, WpEsiTag, WpEsiNewsFunction, WpEsiNewsIndustry, WpEsiNewsRadar, \
WpEsiTechnologyNews, WpEsiTagNews
class News(models.Model):
title = models.CharField(max_length=255, default=None, null=True, blank=True)
url = models.TextField(max_length=2083, default=None, null=True, blank=True)
description = models.TextField(default=None, null=True, blank=True)
......@@ -21,9 +23,9 @@ class News(models.Model):
is_accepted = models.BooleanField(default=False)
radar_id = models.BigIntegerField(default=None, null=True, blank=True)
industry_id = models.CharField(max_length=255,default=None, null=True, blank=True)
function_id = models.CharField(max_length=255,default=None, null=True, blank=True)
technology_id = models.CharField(max_length=255,default=None, null=True, blank=True)
industry_id = models.CharField(max_length=255, default=None, null=True, blank=True)
function_id = models.CharField(max_length=255, default=None, null=True, blank=True)
technology_id = models.CharField(max_length=255, default=None, null=True, blank=True)
tags_id = models.CharField(max_length=255, default=None, null=True, blank=True)
class Meta:
......@@ -115,3 +117,21 @@ class News(models.Model):
)
self.is_accepted = True
self.save()
if self.radar_id:
WpEsiNewsRadar.objects.create(news_id=esi.id, radar_id=self.radar_id)
if self.industry_id:
industry = json.loads(self.industry_id)
WpEsiNewsIndustry.objects.bulk_create([WpEsiNewsIndustry(news_id=esi.id, industry_id=i) for i in industry])
if self.function_id:
func = json.loads(self.function_id)
WpEsiNewsFunction.objects.bulk_create([WpEsiNewsFunction(news_id=esi.id, function_id=i) for i in func])
if self.technology_id:
technology = json.loads(self.technology_id)
WpEsiTechnologyNews.objects.bulk_create(WpEsiTechnologyNews(news_id=esi.id, technology_id=i) for i in technology)
if self.tags_id:
tags = json.loads(self.tags_id)
WpEsiTagNews.objects.bulk_create(WpEsiTagNews(news_id=esi.id, tag_id=i) for i in tags)
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment