get_realizations.py 2.38 KB
#-*- coding:utf-8 -*-

#Copyright (c) 2013, Bartłomiej Nitoń
#All rights reserved.

#Redistribution and use in source and binary forms, with or without modification, are permitted provided 
#that the following conditions are met:

#    Redistributions of source code must retain the above copyright notice, this list of conditions and 
#    the following disclaimer.
#    Redistributions in binary form must reproduce the above copyright notice, this list of conditions 
#    and the following disclaimer in the documentation and/or other materials provided with the distribution.

# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED 
# WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A 
# PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR 
# ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED 
# TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 
# NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 
# POSSIBILITY OF SUCH DAMAGE.
'''
Sript for getting argument realizations.
@author: bartomiej
'''

import codecs

from django.core.management.base import BaseCommand
from django.utils.encoding import smart_str

from dictionary.models import Argument

class Command(BaseCommand):
    args = 'none'
    help = """
    Get list of argument realizations and write it to the real.txt file in
    data dir.
    """

    def handle(self, **options):
        get_realizations()

def get_realizations():
    """
    Get list of argument realizations and write it to the real.txt file in
    data dir.
    """
    real_path = 'data/real.txt'
    real_file = codecs.open(real_path, 'wt', 'utf-8')
    
    for arg in Argument.objects.order_by('type').all():
        if arg.realizations.count() > 0:
            print smart_str(arg.text_rep)
            real_file.write(arg.text_rep + u'-->\n')
            for real in arg.realizations.order_by('type'):
                real_file.write('    ' + real.text_rep + '\n')                                                                   
    real_file.close()