使用 XLRD 导入 Django 递归树

Question

我的数据（电子表格）：

'1',,,
,'1.1',,
,,'1.1.1',
,,'1.1.2',
,,'1.1.3',
,'1.2',,
,'1.3',,
,,'1.3.1',
,,'1.3.2',
,,'1.3.3',
'2',,,
,'2.1',,
,,'2.1.1',
,,,'2.1.1.1'
,,,'2.1.1.2'
,,,'2.1.1.3'

我的模型：

class Vocabulary(models.Model):
    name = CharField(max_length=60)

class Concept(models.Model):
    parent = ForeignKey('self', blank=True, null=True)
    vocabulary = ForeignKey(Vocabulary)
    name = CharField(max_length=60)
    order = IntegerField(default=0)

我想要做什么：

def recurse(sheet):
    'Recurse outer edges of the tree saving concepts.'
        + 'Imply subtree order numbers. There are no numbers in the real data.'

Answer 1

不容易弄清楚，只是分享一下。这就是我使用 Python XLRD 和 Django 将层次结构从 Excel 导入到简单邻接列表树存储的方法。

class XLRDParseError(Exception):
    """The XLS file was malformed."""

def load_xls(fname):
    """Import a hierarchy into the DB from Excel"""
    import xlrd
    xlrd.open_workbook(fname)
    firstSheet = book.sheet_by_index(0)
    v = Vocabulary(title='New Import')
    v.save()
    vid = Vocabulary.objects.get(id=v.id)
    conceptstack = []
    for row in range(firstSheet.nrows):
        blank = 0
        while True:
            cell = firstSheet.cell(row, blank)
            if cell.value:
                break
            blank += 1
        concept = Concept(vocabulary=vid, name=cell.value)
        concept.save()
        if len(conceptstack) < blank:
            raise XLRDParseError
        if len(conceptstack) > blank:
            for i in range(len(conceptstack) - blank):
                conceptstack.pop()
        if conceptstack:
            concept.parent = conceptstack[-1]
            concept.save()
        conceptstack.append(concept)

load_xls('/home/frank/top-navigation.xls')

关键字：层次结构、从 Excel 导入树、导入逗号/制表符分隔的层次结构、从 Excel 导入类别、Python Django XLRD 树导入