from datetime import datetime
from django import forms

from .models import News


class NewsForm(forms.Form):

    def update(self):
        news = News.objects.get(pk=self.data['news_id'])
        news = self._make(news)
        news.save()

    def create(self):
        news = News()
        news = self._make(news)
        news.company_id = self.data['entity']
        news.save()

    def _make(self, news):
        if self.data['title']:
            news.title = self.data['title']
        if self.data['description']:
            news.description = self.data['description']
        if self.data['url']:
            news.url = self.data['url']

        if self.data['radar']:
            news.radar_id = self.data['radar']
        if self.data['industry']:
            print('fdgdfgdfg', self.data['industry'])
            news.industry_id = [int(i) for i in self.data['industry'].split(',')]
        if self.data['function']:
            news.function_id = [int(i) for i in self.data['function'].split(',')]
        if self.data['media']:
            news.media_id = self.data['media']
        if self.data['technology']:
            news.technology_id = [int(i) for i in self.data['technology'].split(',')]
            print(news.technology_id)
        if self.data['type']:
            news.type_id = self.data['type']
        if self.data['region']:
            news.region_id = self.data['region']
        if self.data['date']:
            news.publish_date = datetime.strptime(self.data['date'], '%Y-%m-%d')

        if self.data['tags']:
            print(self.data['tags'])
            news.tags_id = [int(i) for i in self.data['tags'].split(',')]

        return news