Esempio n. 1
0
def import_user():
    import csv
    from stepbystep.models import UserModel, AccountItem, Account
    from stepbystep.libs.tasks import account_crawler
    with open('stepbystep_user.csv', 'r') as users:
        for sid, name, grade, sdut_id, poj_id in csv.reader(
                users.read().splitlines()[1:]):
            if sid and name and grade:
                user = UserModel.objects(username=sid).first()
                if not user:
                    user = UserModel.create_user(username=sid,
                                                 password='******',
                                                 name=name,
                                                 grade=grade)
                if sdut_id:
                    sdut_account = AccountItem.objects.create(origin_oj='sdut',
                                                              username=sdut_id)
                    user.sdut = Account(user_id=sdut_id, account=sdut_account)
                    account_crawler.delay(origin_oj='sdut', username=sdut_id)
                if poj_id:
                    poj_account = AccountItem.objects.create(origin_oj='poj',
                                                             username=poj_id)
                    user.poj = Account(user_id=poj_id, account=poj_account)
                    account_crawler.delay(origin_oj='poj', username=poj_id)
                user.save()
Esempio n. 2
0
def import_user(filename):
    import csv
    from stepbystep.models import UserModel, AccountItem, Account, RoleModel
    from stepbystep.libs.tasks import account_crawler
    sdut_program_1 = RoleModel.objects(name='program_1').first()
    sdut_program_2 = RoleModel.objects(name='program_2').first()
    sdut_data_structure = RoleModel.objects(name='data_structure').first()
    with open(filename, 'r') as users:
        for sid, name, grade, sdut_id, poj_id in csv.reader(
                users.read().splitlines()[1:]):
            if sid and name and grade:
                user = UserModel.objects(
                    username=sid).first()
                if not user:
                    user = UserModel.create_user(
                        username=sid,
                        password='******',
                        name=name,
                        grade=grade
                    )
                if sdut_id:
                    sdut_account = AccountItem.objects.create(
                        origin_oj='sdut',
                        username=sdut_id
                    )
                    user.sdut = Account(
                        user_id=sdut_id,
                        account=sdut_account
                    )
                    user.roles.append(sdut_program_1)
                    user.roles.append(sdut_program_2)
                    user.roles.append(sdut_data_structure)
                    account_crawler.delay(
                        origin_oj='sdut',
                        username=sdut_id
                    )
                if poj_id:
                    poj_account = AccountItem.objects.create(
                        origin_oj='poj',
                        username=poj_id
                    )
                    user.poj = Account(
                        user_id=poj_id,
                        account=poj_account
                    )
                    account_crawler.delay(
                        origin_oj='poj',
                        username=poj_id
                    )
                user.save()
Esempio n. 3
0
def import_user():
    import csv
    from stepbystep.models import UserModel, AccountItem, Account
    from stepbystep.libs.tasks import account_crawler
    with open('stepbystep_user.csv', 'r') as users:
        for sid, name, grade, sdut_id, poj_id in csv.reader(
                users.read().splitlines()[1:]):
            if sid and name and grade:
                user = UserModel.objects(
                    username=sid).first()
                if not user:
                    user = UserModel.create_user(
                        username=sid,
                        password='******',
                        name=name,
                        grade=grade
                    )
                if sdut_id:
                    sdut_account = AccountItem.objects.create(
                        origin_oj='sdut',
                        username=sdut_id
                    )
                    user.sdut = Account(
                        user_id=sdut_id,
                        account=sdut_account
                    )
                    account_crawler.delay(
                        origin_oj='sdut',
                        username=sdut_id
                    )
                if poj_id:
                    poj_account = AccountItem.objects.create(
                        origin_oj='poj',
                        username=poj_id
                    )
                    user.poj = Account(
                        user_id=poj_id,
                        account=poj_account
                    )
                    account_crawler.delay(
                        origin_oj='poj',
                        username=poj_id
                    )
                user.save()
Esempio n. 4
0
    def on_model_change(self, form, model):
        if len(model.password2):
            model.password = UserModel.generate_password(form.password2.data)
        elif not model.password:
            model.password = UserModel.generate_password('12345678')

        if model.bestcoder.user_id:
            account = AccountItem.objects(
                origin_oj='bestcoder',
                username=model.bestcoder.user_id
            ).first()
            if not account:
                account = AccountItem.objects.create(
                    origin_oj='bestcoder',
                    username=model.bestcoder.user_id
                )
            account_crawler.delay(
                origin_oj='bestcoder',
                username=model.bestcoder.user_id
            )
            model.bestcoder.account = account
        else:
            model.bestcoder = Account(user_id='')

        if model.topcoder.user_id:
            account = AccountItem.objects(
                origin_oj='topcoder',
                username=model.topcoder.user_id
            ).first()
            if not account:
                account = AccountItem.objects.create(
                    origin_oj='topcoder',
                    username=model.topcoder.user_id
                )
            account_crawler.delay(
                origin_oj='topcoder',
                username=model.topcoder.user_id
            )
            model.topcoder.account = account
        else:
            model.topcoder = Account(user_id='')

        if model.codeforces.user_id:
            account = AccountItem.objects(
                origin_oj='codeforces',
                username=model.codeforces.user_id
            ).first()
            if not account:
                account = AccountItem.objects.create(
                    origin_oj='codeforces',
                    username=model.codeforces.user_id
                )
            account_crawler.delay(
                origin_oj='codeforces',
                username=model.codeforces.user_id
            )
            model.codeforces.account = account
        else:
            model.codeforces = Account(user_id='')

        if model.sdut.user_id:
            account = AccountItem.objects(
                origin_oj='sdut',
                username=model.sdut.user_id
            ).first()
            if not account:
                account = AccountItem.objects.create(
                    origin_oj='sdut',
                    username=model.sdut.user_id
                )
            account_crawler.delay(
                origin_oj='sdut',
                username=model.sdut.user_id
            )
            model.sdut.account = account
        else:
            model.sdut = Account(user_id='')

        if model.hduoj.user_id:
            account = AccountItem.objects(
                origin_oj='hduoj',
                username=model.hduoj.user_id
            ).first()
            if not account:
                account = AccountItem.objects.create(
                    origin_oj='hduoj',
                    username=model.hduoj.user_id
                )
            account_crawler.delay(
                origin_oj='hduoj',
                username=model.hduoj.user_id
            )
            model.hduoj.account = account
        else:
            model.hduoj = Account(user_id='')

        if model.poj.user_id:
            account = AccountItem.objects(
                origin_oj='poj',
                username=model.poj.user_id
            ).first()
            if not account:
                account = AccountItem.objects.create(
                    origin_oj='poj',
                    username=model.poj.user_id
                )
            account_crawler.delay(
                origin_oj='poj',
                username=model.poj.user_id
            )
            model.poj.account = account
        else:
            model.poj = Account(user_id='')