예제 #1
0
        "c26": return_None(line[30]),
        "c27": return_None(line[31]),
        "d1": check_index(line, 33),
        "u1": check_index(line, 34),
    }

    dataset_args = {}

    for i in args:
        if args[i]:
            dataset_args[i] = args[i]

    if Dataset.objects.filter(varname=line[1], year=line[4]).count():
        p = Dataset.objects.get(varname=line[1], year=line[4])
    else:
        p = Dataset(**args)

    pprint(args)
    for x in args:
        setattr(p, x, args[x])

    # note the missing 24 because we include it everytime
    seznam = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 25, 26, 27]

    old_member = [1, 2, 6, 8, 9, 10, 11, 12, 15, 18, 19, 21, 25, 26, 27]
    new_member = [3, 4, 5, 7, 13, 14, 16, 17, 20, 22, 23]
    picked = [24]

    for i in old_member:
        if getattr(p, "c" + str(i)) == 0.0:
            old_member.remove(i)
예제 #2
0
    def import_data(self):
        wb = xlrd.open_workbook(sys.argv[2])

        for idx in [0, 1]:
            try:
                sh = wb.sheet_by_index(idx)
            except IndexError:
                return

            for row in range(1, sh.nrows):
                line = sh.row_values(row)
                print(line)

                args = {"region":line[0],
                        "varname":line[1],
                        "year":int(line[2]),
                        "descs":line[3],
                        "desce":line[5],
                        "c1":return_None(line[8]),
                        "c2":return_None(line[9]),
                        "c3":return_None(line[10]),
                        "c4":return_None(line[11]),
                        "c5":return_None(line[12]),
                        "c6":return_None(line[13]),
                        "c7":return_None(line[14]),
                        "c8":return_None(line[15]),
                        "c9":return_None(line[16]),
                        "c10":return_None(line[17]),
                        "c11":return_None(line[18]),
                        "c12":return_None(line[19]),
                        "c13":return_None(line[20]),
                        "c14":return_None(line[21]),
                        "c15":return_None(line[22]),
                        "c16":return_None(line[23]),
                        "c17":return_None(line[24]),
                        "c18":return_None(line[25]),
                        "c19":return_None(line[26]),
                        "c20":return_None(line[27]),
                        "c21":return_None(line[28]),
                        "c22":return_None(line[29]),
                        "c23":return_None(line[30]),
                        "c24":return_None(line[31]),
                        "c25":return_None(line[32]),
                        "c26":return_None(line[33]),
                        "c27":return_None(line[34]),
                        "c28":return_None(line[35]),
                        # "d1": check_index(line, 36),
                        # "u1": check_index(line, 37),
                        }

                dataset_args = {}

                for i in args:
                  if args[i]:
                    dataset_args[i] = args[i]

                if Dataset.objects.filter(varname=args['varname'], year=args['year']).exists():
                    p = Dataset.objects.get(varname=args['varname'], year=args['year'])
                else:
                    p = Dataset(**args)

                for x in args:
                   setattr(p, x, args[x])
                p.save()
                p.generate_sample()