예제 #1
0
def import_data(in_file='quick_export.csv'):
#    reader = csv.reader(open(os.path.join(DEFAULT_PATH, in_file)))
    reader = csv.reader(open(in_file))
    for i, line in enumerate(reader):
        if not i:
            continue
        uid = line[1]
        category = line[5].decode('utf-8')
        subcategory = line[6].decode('utf-8')

        if category:
            category_obj = Category.all().filter('name =', category)
            if category_obj.count():
                category_obj = category_obj[0]
            else:
                category_obj = Category(name=category)
                category_obj.put()
        else:
            category_obj = None

        if category and subcategory:
            group_obj = Group.all().filter('name =', subcategory).filter('category =', category_obj)
            if group_obj.count():
                group_obj = group_obj[0]
            else:
                group_obj = Group(name=subcategory, category=category_obj)
                group_obj.put()
        else:
            group_obj = None


        gift = Gift.all().filter('uid =', uid)

        if gift.count():
            gift = gift[0]
            gift.category = category_obj
            gift.group = group_obj
            gift.put()
            print 'Gift %s updated' % uid
        else:
            print 'Gift %s not found' % uid
예제 #2
0
파일: views.py 프로젝트: gmist/five-studio
def update_gifts(task_id, create=False, except_fields=EXCEPT_FIELDS_UPDATE):
    set_obj = memcache.get(task_id)
    flag = False
    if set_obj:
        data = simplejson.loads(set_obj)
        memcache.delete(task_id)
        id_1c = data.get('id_1c', '')
        logging.info("API update 1c: %s" % id_1c)
        name = data.get('name', '')
        logging.info("API update name: %s" % name)
        if not id_1c and not name:
            if not create:
                return render_to_response('empty.html')
        if id_1c:
            gifts_obj = Gift.all().filter('id_1c =', id_1c)
            if not gifts_obj.count():
                gifts_obj = Gift.all().filter('name =', name)
        else:
            gifts_obj = Gift.all().filter('name =', name)
        if gifts_obj.count() and create:
            return render_to_response('empty.html')
        gift = None
        if gifts_obj.count() and not create:
            gift = gifts_obj[0]
        if not gifts_obj.count() and create:
            gift = Gift(name=name)
        if gift is None:
            return render_to_response('empty.html')

        # update category and groups
        category = data.get('subcategory', '-')
        group = data.get('group', '-')
        try:
            a = gift.category.name
            b = gift.group.name
        except Exception:
            gift.category = gift.group = None

        if category != '-' and group != '-':
            if not category or not group:
                if gift.category or gift.group:
                    gift.category = None
                    gift.group = None
                    flag = True
            else:
                category_obj = Category.all().filter('name =', category)
                if category_obj.count():
                    category_obj = category_obj[0]
                else:
                    category_obj = Category(name=category)
                    category_obj.put()
                if not gift.category or gift.category != category_obj:
                    gift.category = category_obj
                    flag = True

                group_obj = Group.all().filter(
                    'category =', category_obj).filter('name =', group)
                if group_obj.count():
                    group_obj = group_obj[0]
                else:
                    group_obj = Group(name=group, category=category_obj)
                    group_obj.put()
                if not gift.group or gift.group != group_obj:
                    gift.group = group_obj
                    flag = True

        # and other fields
        for key in data.keys():
            if key in except_fields:
                continue
            value = data.get(key, None)
            if value is None:
                continue
            if key == 'receipt_date':
                try:
                    new_date = datetime.date(
                        datetime.strptime(value, '%Y-%m-%d'))
                    if new_date == gift.receipt_date:
                        continue
                    gift.receipt_date = new_date
                    flag = True
                except ValueError:
                    if gift.receipt_date is not None:
                        gift.receipt_date = None
                        flag = True
                continue
            try:
                old_val = getattr(gift, key)
                if old_val == value:
                    continue
                setattr(gift, key, value)
                flag = True
            except AttributeError:
                pass
        if flag:
            gift.put()
            gift_views.get_gift(gift.uid, True)
    return render_to_response('empty.html')