# -*- coding: utf-8 -*- import copy import calendar as python_calendar from pytils.dt import ru_strftime from django.conf import settings from django.db.models import Q from django.utils.translation import ugettext as _ from django.utils.translation import get_language from hvad.utils import get_translation, set_cached_translation, get_cached_translation from functions.translate import fill_with_signal from service.models import Service from photologue.models import Gallery class ExpoMixin(object): def get_logo(self): logo = self.files.filter(purpose='logo') if logo: return logo[0] return logo def get_preview(self): preview = self.files.filter(purpose='preview') if preview: return preview[0] return preview def get_photos(self): photos = self.files.filter(purpose='photo') return photos def get_dates(data_begin, data_end): lang = get_language() if data_end and data_begin: delta = data_end - data_begin if delta.days > 28 and data_end.month == data_begin.month: return data_begin.strftime('%B') elif delta.days < 1: if lang == 'ru': return ru_strftime("%d %B %Y", date=data_begin, inflected=True) return data_begin if data_begin.month != data_end.month: if lang == 'ru': start = ru_strftime("%d %B", date=data_begin, inflected=True) else: start = u'{month} {day}'.format( month=data_begin.strftime('%B'), day=data_begin.day) else: start = data_begin.day if lang == 'ru': end = ru_strftime("%d %B", date=data_end, inflected=True) else: end = u'{month} {day}'.format( month=data_end.strftime('%B'), day=data_end.day) return _(u'c {start} по {end}').format(start=start, end=end) if lang == 'ru': return ru_strftime("%d %B %Y", date=data_begin, inflected=True) return data_begin class EventMixin(object): @property def get_dates(self): return get_dates(self.data_begin, self.data_end) @property def get_place_name(self): tr = get_cached_translation(self.place) if tr is None: tr = get_translation(self.place) set_cached_translation(self.place, tr) return self.place.name def get_permanent_url(self): url = '%s%s/'%(self.get_catalog_url(), self.url) return url def get_main_link(self): if self.main: return self.main.link.get_click_link() else: return self.get_permanent_url() def get_paid_catalog_url(self): return self.paid_new.catalog.get_click_link() def org_split(self): if self.org: return self.org.split(';') else: return [] def get_logo(self): return self.logo def get_preview(self): preview = self.files.filter(purpose='preview') if preview: return preview[0] return preview def get_photos(self): if self.photogallery: return self.photogallery.photos.language().all() else: return None """ photos = self.files.filter(purpose='photo') return photos """ def on(self): self.is_published = True self.canceled_by_administrator = False self.save() def off(self): self.is_published = False self.canceled_by_administrator = True self.save() def cancel(self): self.canceled_by_administrator = True def get_services(self): country_ids = [item for item, bool in self.country.services if bool==True] ids = [item for item, bool in self.services if bool==True and item in country_ids] return list(Service.objects.language().filter(url__in=ids).order_by('sort')) def get_services_detail(self, excluded, _type): if not isinstance(getattr(self, '_get_services_detail', None), list): # excluded = ['visit', 'tickets'] # country_ids = [item for item, bool in self.country.services if bool==True] services = [item for item, bool in self.services if bool==True] qs = Service.objects.language() if excluded is not None: qs = qs.exclude(url__in=excluded) # qs = qs.filter(Q(Q(url__in=country_ids) & Q(type=_type)) | Q(url__in=ids)) qs = qs.filter(url__in=services) self._get_services_detail = list(qs) # import pdb; pdb.set_trace() #двигаем билеты сразу за переводом if excluded is None or 'tickets' not in excluded: translator_idx = tickets = None for idx, service in enumerate(self._get_services_detail): if service.url == 'translator': translator_idx = idx + 1 elif service.url == 'tickets': tickets = service if tickets and translator_idx: self._get_services_detail.remove(tickets) self._get_services_detail.insert(translator_idx, tickets) return self._get_services_detail def duration_days(self, month=None): if not month: d = self.data_end - self.data_begin return d.days+1 else: if self.data_begin.month == month and self.data_end.month ==month: d = self.data_end - self.data_begin return d.days + 1 if self.data_begin.month == month: last_day = python_calendar.monthrange(self.data_begin.year, month)[1] return last_day - self.data_begin.day + 1 if self.data_end.month == month: return self.data_end.day return 0 def upload_photo(self, photo, gallery=None): """ uploading photo to gallery """ if gallery is None: gallery = self.get_gallery() gallery.photos.add(photo) def get_gallery(self): if self.photogallery: return self.photogallery data = {} model = type(self) for code, name in settings.LANGUAGES: # access to translated fields obj = model._meta.translations_model.objects.get(language_code=code, master__id=self.id) #access to translated fields data['title_%s' % code] = obj.name data['description_%s ' % code] = obj.description gallery = Gallery() fill_with_signal(Gallery, gallery, data) self.photogallery = gallery self.save() return gallery def tags(self): return self.tag.language().all() def copy(self, url): """ Copy event with new url :param url: new url for event :return: event object """ # check url Model = type(self) try: Model.objects.get(url=url) return _(u'Событие с таким урлом уже существует') except Model.DoesNotExist: pass duplicate = copy.copy(self) duplicate.url = url # Setting pk to None. for Django its a new object. duplicate.pk = None # copy translations ignore_fields = ['id', 'master', 'language_code'] duplicate.translate('ru') tr = self._meta.translations_model.objects.get(language_code='ru', master__id=self.pk) for field in duplicate._translated_field_names: if field in ignore_fields: continue setattr(duplicate, field, getattr(tr, field)) duplicate.is_published = False duplicate.save() # save but lost all ManyToMany relations # copy relations for field in self._meta.many_to_many: source = getattr(self, field.attname) destination = getattr(duplicate, field.attname) for item in source.all(): destination.add(item) return duplicate class GetURLorPK(object): @property def url_or_pk(self): return self.url or self.pk