# coding=utf8
import os
from django.core.wsgi import get_wsgi_application
os.environ['DJANGO_SETTINGS_MODULE'] = 'web.local_settings'
application = get_wsgi_application()
from django.contrib.auth.models import User
try:
from student.models import Profile
except:
from user_info.models import UserProfile as Profile
def clean_line(text):
# s.strip(rm) 删除s字符串中开头、结尾处
# 当rm为空时，默认删除空白符（包括"\n", "\r", "\t", " ")
# split用法:按某一个字符分割，如‘.’,‘(空格) ’
return [i for i in text.strip().split(' ') if i]
def read_text(): # 读数据
f = open("/home/yangheng/下载/test.txt", 'r')
# .readlines()一次读取整个文件。
lines = f.readlines()
cleaned_text = [clean_line(line) for line in lines if line.strip()]
for line in cleaned_text:
create_user(*line)
def create_user(username, email='', password='123456', name=''):
name = name or username
email = email or '%[email protected]' % username
print 'args:', username, email, password, name
user, created = User.objects.get_or_create(username=username)
if created:
user.email = email
user.set_password(password)
# user.is_active = True edx中需要加入
user.save()
profile, created = Profile.objects.get_or_create(user=user)
profile.name = name
profile.save()
else:
print '%s username is exist' % username
def main():
read_text()
main()
