如何将扫描文本中的数据导入Django模型

时间:2011-10-21 03:24:55

标签: python django import

我有数百页的“测验”问题,多项选择以及相关的答案键和解释。我正在尝试创建一个简单的Django应用程序来管理这些问题。我创建了一个简单但有效的Python解析器,用于将扫描的OCR页面解析为适当的对象。

我希望有一个“实用程序”来启用此Django应用程序的管理员将测试内容从OCR'd文件导入相关的Django数据库表。这将是一项罕见的任务,并且不一定适合包含在Web UI中。

我问过使用中间JSON / YAML灯具,并被告知更合适的方法是直接创建和保存我的模型实例[1]。然后我尝试按照[2]和[3]建议的方式创建一个独立的脚本,但是无法克服 kwargs = {"app_label": model_module.__name__.split('.')[-2]} IndexError: list index out of range 错误。

我也遇到过[4]关于创建自定义django-admin.py/manage.py命令的问题。这似乎是处理任务的逻辑上合适的方式;但是,我很乐意听到那些有更多经验和大脑的人(我吃过我的全部:)。

参考文献:

  1. Importing data from scanned text into Django as YAML fixture or SQL
  2. what is the simplest way to create a table use django db api ,and base on 'Standalone Django scripts'
  3. Standalone Scripts
  4. Writing custom django-admin commands
  5. 示例:

    • OCR'd Text
      

    第12页   34.Hiedegger是_____。 (a)哲学家(b)嘘乞丐(c)a和b(d)这些都不是...... ...

    • Django模型

      class Question(models.Model):
          text = models.TextField()
      class Choice(models.Model):
          question = models.ForeignKey(Question)
          order = models.IntegerField(default=1)
          text = models.TextField()
      
    • 目标,像这样...

      q = Question.objects.create(text="Hiedegger is a _____ .")
      q.save()
      c = Choice(text="philosopher", order=1, question=q.pk)
      c.save()
      

2 个答案:

答案 0 :(得分:0)

这是我提出的工作版本。脏,但有效。 @akonsu和@Ivan Kharlamov都很有帮助。感谢...

import os, re, Levenshtein as lev, codecs
from SimpleQuiz.quiz.models import Choice, Question
from django.core.management.base import BaseCommand, CommandError
import optparse

class Command(BaseCommand):
    args = '--datapath=/path/to/text/data/'
    can_import_settings = True
    help = 'Imports scanned text into Questions and Choices'
    option_list = BaseCommand.option_list + (
        optparse.make_option('--datapath', action='store', type='string',
                             dest='datapath',
                             help='Path to OCRd text files to be parsed.'),
        )
    requires_model_validation = True
    # Parser REs
    BACKUP_RE = re.compile(r'\~$|bak$|back$|backup$')
    QUEST_RE = re.compile(r'^[0-9]{1,3}[.][ ]')
    CHOICE_RE = re.compile(r'^[a-e][.][ ]')

    def handle(self, *args, **options):
        # get the data path
        try:
            os.path.exists(options['datapath'])
        except Exception as e:
            raise CommandError("None or invalid path provided: %s" % e.message)
        self.datapath = os.path.expanduser(options['datapath'])

        # generate list of text strings from lines in target files
        self.data_lines = []
        for fn in os.listdir(os.path.join(self.datapath, 'questions/')):
            if self.BACKUP_RE.search(fn):
                self.stderr.write("Skipping backup: %s\n" % (fn))
            else:
                for line in codecs.open(os.path.join(self.datapath, 'questions/', fn), 'r', encoding='latin-1'):
                    if not self.is_boilerplate(line):
                        if not line.strip() == '':
                            self.data_lines.append(line)

    #----------------------------------------------------------------------- 
    #--------------------- Parse the text lines and create Questions/Choices
    #----------------------------------------------------------------------- 
        cur_quest = None
        cur_choice = None
        cur_is_quest = False
        questions = {}
        choices = {}
        for line in self.data_lines:
            if self.is_question(line):
                [n, txt] = line.split('.', 1)
                qtext = txt.rstrip() + " "
                q = Question.objects.create(text=qtext)
                q.save()
                cur_quest = q.pk
                questions[cur_quest] = q
                cur_is_quest = True
            elif self.is_choice(line):
                [n, txt] = line.split('.', 1)
                num = self.char2dig(n)
                ctext = txt.rstrip() + " "
                c = Choice.objects.create(text=ctext, order=num, question=questions[cur_quest])
                c.save()
                cur_choice = c.pk
                choices[cur_choice] = c
                cur_is_quest = False
            else:
                if cur_is_quest:
                    questions[cur_quest].text += line.rstrip() + " "
                    questions[cur_quest].save()
                else:
                    choices[cur_choice].text += line.rstrip() + " "
                    choices[cur_choice].save()
        self.stdout.write("----- FINISHED -----\n")
        return None

    def is_question(self, arg_str):
        if self.QUEST_RE.search(arg_str):
            return True
        else:
            return False

    def is_choice(self, arg_str):
        if self.CHOICE_RE.search(arg_str):
            return True
        else:
            return False

    def char2dig(self, x):
        if x == 'a':
            return 1
        if x == 'b':
            return 2
        if x == 'c':
            return 3
        if x == 'd':
            return 4
        if x == 'e':
            return 5

    def is_boilerplate(self, arg_str):
        boilerplate = [u'MFT PRACTICE EXAMINATIONS',
                       u'BERKELEY TRAINING ASSOCIATES ' + u'\u00A9' + u' 2009',
                       u'BERKELEY TRAINING ASSOCIATES',
                       u'MARRIAGE AND FAMILY THERAPY',
                       u'PRACTICE EXAMINATION 41',
                       u'Page 0', u'Page 1', u'Page 2', u'Page 3', u'Page 4',
                       u'Page 5', u'Page 6', u'Page 7', u'Page 8', u'Page 9',
                       ]
        for bp in boilerplate:
            if lev.distance(bp.encode('utf-8'), arg_str.encode('utf-8')) < 4:
                return True
        return False

答案 1 :(得分:0)

  

然后我尝试按照建议的方式创建一个独立的脚本   [2]和[3]但是无法克服kwargs = {“app_label”:   model_module。 name .split('。')[ - 2]} IndexError:列出索引   范围错误。

我有相同的列表索引错误。这是由我在脚本中导入模型的方式引起的。我曾经这样做过:

from models import Table1, Table2

然后我意识到Python脚本不是应用程序的一部分,因此我将导入更改为:

from myapp.models import Table1, Table2

我的Python脚本是使用以下shell脚本启动的:

export DJANGO_SETTINGS_MODULE=settings
export PYTHONPATH=/path/to/my/site
python myscript.py "$@"