# -*- coding: utf-8 -*-

import hashlib

import xlrd
from pysnippets.strsnippets import strip
from TimeConvert import TimeConvert as tc

from tamron.models import TamronRegisterInfo, TamronUserInfo


def pre_user():
    TamronUserInfo.objects.all().delete()

    workbook = xlrd.open_workbook('./tamron/data/LRC.xls')
    # sheets = workbook.sheets()
    # sheet = sheets[0]
    sheet = workbook.sheet_by_name('User')
    nrows = sheet.nrows

    for idx in range(1, nrows):
        rvals = sheet.row_values(idx)
        print rvals

        phone = str(int(strip(rvals[2]))) if strip(rvals[2]) else ''

        TamronUserInfo.objects.create(
            user=strip(rvals[0]),
            sex=strip(rvals[3]),
            zhiwei=strip(rvals[4]),
            tel=phone,
            email=strip(rvals[1]),
            password=hashlib.md5(phone).hexdigest(),
            pinpai=strip(rvals[5]),
            xinghao=strip(rvals[6]),
            time=strip(rvals[7]),
            gzleixing=strip(rvals[8]),
            gxtedian=strip(rvals[9]),
            ctime=tc.string_to_local_timestamp(strip(rvals[10]), format='%Y-%m-%d'),
        )


def pre_reg():
    TamronRegisterInfo.objects.filter(fid=TamronRegisterInfo.BRAND).delete()
    TamronRegisterInfo.objects.filter(fid=TamronRegisterInfo.MODEL).delete()
    TamronRegisterInfo.objects.filter(fid=TamronRegisterInfo.METIER).delete()

    workbook = xlrd.open_workbook('./tamron/data/LRC.xls')
    # sheets = workbook.sheets()
    # sheet = sheets[0]
    sheet = workbook.sheet_by_name('User')
    nrows = sheet.nrows

    brands = set()
    models = set()
    metiers = set()

    for idx in range(1, nrows):
        rvals = sheet.row_values(idx)
        print rvals

        brand = strip(rvals[5])
        if brand:
            brands.add(brand)

        model = strip(rvals[6])
        if model:
            models.add(model)

        metier = strip(rvals[4])
        if metier:
            metiers.add(metier)

    for brand in brands:
        TamronRegisterInfo.objects.create(fid=TamronRegisterInfo.BRAND, title=brand)

    for model in models:
        TamronRegisterInfo.objects.create(fid=TamronRegisterInfo.MODEL, title=model)

    for metier in metiers:
        TamronRegisterInfo.objects.create(fid=TamronRegisterInfo.METIER, title=metier)