src/hdalab/services.py
author ymh <ymh.work@gmail.com>
Tue, 05 Apr 2016 19:52:17 +0200
changeset 680 541cd2de37d8
parent 618 cb8b833ad122
child 686 385e3a12ee27
permissions -rw-r--r--
Added tag V03.02 for changeset 1911af6da33f
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
458
604b887e70c3 add state history to renkan, correct get into post, generally prepare ground for mail management
ymh <ymh.work@gmail.com>
parents:
diff changeset
     1
# -*- coding: utf-8 -*-
604b887e70c3 add state history to renkan, correct get into post, generally prepare ground for mail management
ymh <ymh.work@gmail.com>
parents:
diff changeset
     2
'''
604b887e70c3 add state history to renkan, correct get into post, generally prepare ground for mail management
ymh <ymh.work@gmail.com>
parents:
diff changeset
     3
Created on Feb 22, 2015
604b887e70c3 add state history to renkan, correct get into post, generally prepare ground for mail management
ymh <ymh.work@gmail.com>
parents:
diff changeset
     4
604b887e70c3 add state history to renkan, correct get into post, generally prepare ground for mail management
ymh <ymh.work@gmail.com>
parents:
diff changeset
     5
@author: ymh
604b887e70c3 add state history to renkan, correct get into post, generally prepare ground for mail management
ymh <ymh.work@gmail.com>
parents:
diff changeset
     6
'''
462
df8b3b49e81c correct bug and add email sending
ymh <ymh.work@gmail.com>
parents: 458
diff changeset
     7
import logging
492
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
     8
import os
462
df8b3b49e81c correct bug and add email sending
ymh <ymh.work@gmail.com>
parents: 458
diff changeset
     9
492
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    10
from django.conf import settings
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    11
from django.core.urlresolvers import reverse
458
604b887e70c3 add state history to renkan, correct get into post, generally prepare ground for mail management
ymh <ymh.work@gmail.com>
parents:
diff changeset
    12
from django.db import transaction
462
df8b3b49e81c correct bug and add email sending
ymh <ymh.work@gmail.com>
parents: 458
diff changeset
    13
458
604b887e70c3 add state history to renkan, correct get into post, generally prepare ground for mail management
ymh <ymh.work@gmail.com>
parents:
diff changeset
    14
from hdalab.models.renkan import HdalabRenkanStateTransition
604b887e70c3 add state history to renkan, correct get into post, generally prepare ground for mail management
ymh <ymh.work@gmail.com>
parents:
diff changeset
    15
462
df8b3b49e81c correct bug and add email sending
ymh <ymh.work@gmail.com>
parents: 458
diff changeset
    16
492
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    17
from subprocess import check_call
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    18
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    19
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    20
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    21
462
df8b3b49e81c correct bug and add email sending
ymh <ymh.work@gmail.com>
parents: 458
diff changeset
    22
logger = logging.getLogger(__name__)
df8b3b49e81c correct bug and add email sending
ymh <ymh.work@gmail.com>
parents: 458
diff changeset
    23
458
604b887e70c3 add state history to renkan, correct get into post, generally prepare ground for mail management
ymh <ymh.work@gmail.com>
parents:
diff changeset
    24
@transaction.atomic
604b887e70c3 add state history to renkan, correct get into post, generally prepare ground for mail management
ymh <ymh.work@gmail.com>
parents:
diff changeset
    25
def change_renkan_state(hda_renkan, state, message=None, author=None):
604b887e70c3 add state history to renkan, correct get into post, generally prepare ground for mail management
ymh <ymh.work@gmail.com>
parents:
diff changeset
    26
    
604b887e70c3 add state history to renkan, correct get into post, generally prepare ground for mail management
ymh <ymh.work@gmail.com>
parents:
diff changeset
    27
    if state != hda_renkan.state:
604b887e70c3 add state history to renkan, correct get into post, generally prepare ground for mail management
ymh <ymh.work@gmail.com>
parents:
diff changeset
    28
        
604b887e70c3 add state history to renkan, correct get into post, generally prepare ground for mail management
ymh <ymh.work@gmail.com>
parents:
diff changeset
    29
        HdalabRenkanStateTransition.objects.create(renkan=hda_renkan, from_state=hda_renkan.state, to_state=state, message=message, author=author)
604b887e70c3 add state history to renkan, correct get into post, generally prepare ground for mail management
ymh <ymh.work@gmail.com>
parents:
diff changeset
    30
        hda_renkan.state = state
462
df8b3b49e81c correct bug and add email sending
ymh <ymh.work@gmail.com>
parents: 458
diff changeset
    31
        hda_renkan.save()
df8b3b49e81c correct bug and add email sending
ymh <ymh.work@gmail.com>
parents: 458
diff changeset
    32
        
492
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    33
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    34
def renkan_capture_preview(hdalab_renkan):
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    35
    #get last state date or last modification date
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    36
    #states are ordered by ts
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    37
    folder_date = hdalab_renkan.renkan.modification_date
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    38
    states = hdalab_renkan.states
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    39
    state = None
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    40
    if states.count() > 0:
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    41
        state = states.all()[0]
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    42
    if state:
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    43
        folder_date =  state.ts
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    44
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    45
    rel_export_path_dir = "thumbnails/renkan/%04d/%02d/%02d" % (folder_date.year, folder_date.month, folder_date.day)
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    46
    export_path_dir = os.path.join(settings.MEDIA_ROOT,rel_export_path_dir)
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    47
    export_filename = "%s.png" % hdalab_renkan.renkan.rk_id
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    48
    
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    49
    export_path = os.path.join(export_path_dir, export_filename)
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    50
    rel_export_path = os.path.join(rel_export_path_dir, export_filename)
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    51
    
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    52
    if not os.path.exists(export_path_dir):
618
cb8b833ad122 correct celery errors
ymh <ymh.work@gmail.com>
parents: 492
diff changeset
    53
            try:
cb8b833ad122 correct celery errors
ymh <ymh.work@gmail.com>
parents: 492
diff changeset
    54
                os.makedirs(export_path_dir)
cb8b833ad122 correct celery errors
ymh <ymh.work@gmail.com>
parents: 492
diff changeset
    55
            except OSError, e:
cb8b833ad122 correct celery errors
ymh <ymh.work@gmail.com>
parents: 492
diff changeset
    56
                if e.errno != 17:
cb8b833ad122 correct celery errors
ymh <ymh.work@gmail.com>
parents: 492
diff changeset
    57
                    raise   
cb8b833ad122 correct celery errors
ymh <ymh.work@gmail.com>
parents: 492
diff changeset
    58
                # time.sleep might help here
cb8b833ad122 correct celery errors
ymh <ymh.work@gmail.com>
parents: 492
diff changeset
    59
                pass
cb8b833ad122 correct celery errors
ymh <ymh.work@gmail.com>
parents: 492
diff changeset
    60
            
492
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    61
    preview_dim = getattr(settings, 'RENKAN_PREVIEW_DIM', (500,500))
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    62
    preview_wait = getattr(settings, 'RENKAN_PREVIEW_WAIT', 5000)
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    63
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    64
    preview_args = [
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    65
        getattr(settings,'RENKAN_PREVIEW_PHANTOMJS_PATH','phantomjs'),
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    66
        os.path.join(os.path.dirname(os.path.abspath(__file__)),'scripts/capture-phantomjs.js'),
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    67
        "%s%shdalab%s?rk_id=%s" % (settings.WEB_URL, settings.BASE_URL, reverse('renkan_full'),hdalab_renkan.renkan.rk_id),
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    68
        export_path,
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    69
        "--width=%d" % preview_dim[0],
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    70
        "--height=%d" % preview_dim[1],
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    71
        "--wait=%d" % preview_wait
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    72
    ]
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    73
    check_call(preview_args)
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    74
    
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    75
    hdalab_renkan.renkan.image = rel_export_path
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    76
    hdalab_renkan.renkan.save()
19220d52bce7 add preview calculation
ymh <ymh.work@gmail.com>
parents: 462
diff changeset
    77