def home():
print "config.AUTHENTICATE {}".format(config.AUTHENTICATE)
if config.AUTHENTICATE:
return render_template('index.html',
login_form=LoginForm(),
create_form=CreateAccountForm())
else:
email = 'default'
password = '******'
access_key_id = config.AWS_ID
secret_access_key = config.AWS_KEY
aws_bucket_name = config.AWS_BUCKET_NAME
next_backend_global_host = config.NEXT_BACKEND_GLOBAL_HOST
if not User.objects(email=email):
user = User(email=email,
access_key_id=access_key_id,
secret_access_key=secret_access_key,
aws_bucket_name=aws_bucket_name,
next_backend_global_host=next_backend_global_host)
user.set_password(password)
user.save()
else:
user = User.objects(email=email).first()
login_user(user, remember=True)
return redirect(url_for('dashboard._dashboard'))
def saveUser(self, _param):
"""
保存用户信息
@param _param:
@return:
"""
flag_add = True
user = User.objects.filter(
pk=_param['id']).first() if _param.get('id', None) else None
if user:
flag_add = False
if user and User.objects.exclude(pk=_param['id']).filter(
username=_param['username']).count():
raise ex(u'该用户名已经存在!')
if not user:
if _param.get('username', None) and User.objects.filter(
username=_param['username']):
raise ex(u'该用户名已经存在!')
user = User()
dic2obj(user, [
'username', 'truename', 'name', 'email', 'phone', 'department_id',
'role_id'
], _param)
if _param.get('password', ''):
user.password = _param['password']
user.save()
self.logAtion(ActionLog.ACTION_MODIFY, User, user.id)
def signup():
user = User()
form = UserForm()
if form.validate_on_submit():
form.populate_obj(user)
user.save(force_insert=True)
flash('Your account have been created')
return redirect(url_for('index'))
return render_template('index.html', form=form)
def submit(request):
try:
email = request.POST.get('email')
locale_id = request.POST.get('locale_id')
user = User(email=email.lower(), locale_id_id=locale_id, active=True)
user.save()
return HttpResponseRedirect(reverse('subscription:submitted'))
except IntegrityError as err:
log_error(err)
return HttpResponseRedirect(reverse('subscription:nosubmit'))
def init_data():
"""Fish data for project"""
if prompt_bool('Do you want to kill your db?'):
db.drop_all()
try:
db.create_all()
except:
pass
user = User.query.filter(User.email=='*****@*****.**').first()
if user is None:
user = User(username='******', email='*****@*****.**', password='******')
user.save()
def __init__(self):
global _player
if _player is not None:
raise "Player was initialized!"
random.seed(time.clock())
print("Starting player...")
self.vlcInstance = vlc.Instance(
"--live-caching=0 --network-caching=0 --norm-buff-size=8")
self.vlcPlayer = self.vlcInstance.media_player_new()
self.vlcEvents = self.vlcPlayer.event_manager()
self.vlcEvents.event_attach(vlc.EventType.MediaPlayerEndReached,
self.songFinished)
self.vlcEvents.event_attach(vlc.EventType.MediaPlayerEncounteredError,
self.songCorrupted)
self.vlcEvents.event_attach(vlc.EventType.MediaStateChanged,
self.mediaStateChanged)
# Import old DJPajton playlist
if Song.objects.count() == 0 and os.path.isfile('music4you.playlist'):
with open("music4you.playlist", "r") as f:
playlist = json.loads(f.read())
for item in playlist:
print("Item: {}".format(item))
song = Song()
song.id = int(item['id'])
song.date = item['addedDate']
song.active = (bool(item['deleted']) == False)
song.url = item['url']
song.title = item['title']
try:
user = User.objects.get(login=item['addedLogin'])
except:
user = User()
user.active = False
user.type = USER_TYPE_SKYPE
user.login = item['addedLogin']
user.displayName = user.login
user.save()
song.user = user
song.save()
#self.vlcPlayer.audio_set_volume(100)
if Song.objects.count() == 0:
return
def form_valid(self, form):
# Create a user, but remember to set inactive!
user = User()
user.username = form.cleaned_data['email']
user.email = form.cleaned_data['email']
user.is_active = False
try:
user.save()
except IntegrityError:
form.add_error('email',
gettext('Shop with this email already exists.'))
return super(ShopRegisterView, self).form_invalid(form)
self.object = form.save(commit=False)
self.object.postcode = Postcode.objects.get(
postcode=form.cleaned_data['postcode_special'])
self.object.user = user
self.object.save()
current_site = get_current_site(self.request)
context = {
'shopname': self.object.name,
'user': user,
'domain': current_site.domain,
'uid': urlsafe_base64_encode(force_bytes(user.pk)),
'token': account_activation_token.make_token(user),
}
html_message = render_to_string('emails/account_activation.html',
context)
txt_message = render_to_string('emails/account_activation.txt',
context)
email = EmailMultiAlternatives(gettext('FOODBEE - Confirm email'),
txt_message)
email.from_email = settings.DEFAULT_FROM_EMAIL
email.to = [self.object.email]
email.attach_alternative(html_message, "text/html")
email.content_subtype = 'html'
email.mixed_subtype = 'related'
with open('base/static/base/img/fb_logo.png', mode='rb') as f:
image = MIMEImage(f.read())
image.add_header('Content-ID', "<Foodbee_logo_long.png>")
email.attach(image)
email.send()
return super().form_valid(form)
class KitTestCase(TestCase):
def create_app(self):
return AppFactory(TestingConfig).get_app(__name__)
def setUp(self):
db.create_all()
self.user = User(username='******', email='*****@*****.**', password='******')
self.user.save()
def tearDown(self):
db.session.remove()
db.drop_all()
def assertContains(self, response, text, count=None,
status_code=200, msg_prefix=''):
"""
Asserts that a response indicates that some content was retrieved
successfully, (i.e., the HTTP status code was as expected), and that
``text`` occurs ``count`` times in the content of the response.
If ``count`` is None, the count doesn't matter - the assertion is true
if the text occurs at least once in the response.
"""
if msg_prefix:
msg_prefix += ": "
self.assertEqual(response.status_code, status_code,
msg_prefix + "Couldn't retrieve content: Response code was %d"
" (expected %d)" % (response.status_code, status_code))
real_count = response.data.count(text)
if count is not None:
self.assertEqual(real_count, count,
msg_prefix + "Found %d instances of '%s' in response"
" (expected %d)" % (real_count, text, count))
else:
self.assertTrue(real_count != 0,
msg_prefix + "Couldn't find '%s' in response" % text)
def create():
create_form = CreateAccountForm(request.form)
if create_form.validate_on_submit():
print "form data",create_form.new_email.data
new_email = create_form.new_email.data
new_password = create_form.new_password.data
confirm_password = create_form.confirm_password.data
print User.objects()
if User.objects(email=new_email).count() == 0:
if new_password == confirm_password:
# Set the hashed password on the user.
user = User(email=new_email)
user.set_password(new_password)
user.save()
login_user(user, remember=True)
return redirect(url_for('dashboard._dashboard'))
else:
flash("Passwords don't match!", "create_error")
return redirect('#create')
else:
flash("Username exists!", "create_error")
return redirect('#create')
# login form is not valid
return render_template('index.html', login_form=LoginForm(), create_form=create_form)
def create():
create_form = CreateAccountForm(request.form)
if create_form.validate_on_submit():
print "form data",create_form.new_email.data
new_email = create_form.new_email.data
new_password = create_form.new_password.data
confirm_password = create_form.confirm_password.data
print User.objects()
if User.objects(email=new_email).count() == 0:
if new_password == confirm_password:
# Set the hashed password on the user.
user = User(email=new_email)
user.set_password(new_password)
user.save()
login_user(user, remember=True)
return redirect(url_for('dashboard._dashboard'))
else:
flash("Passwords don't match!", "create_error")
return redirect('#create')
else:
flash("Username exists!", "create_error")
return redirect('#create')
# login form is not valid
return render_template('index.html', login_form=LoginForm(), create_form=create_form)
class UserTests(TestCase):
def setUp(self):
self.client = Client()
self.user1 = User(username="******",
first_name="Fa",
password="******")
self.user1.save()
self.user2 = User(username="******",
first_name="Fa",
password="******")
self.user2.save()
def test_get_users(self):
user_list = self.client.get('/users/')
u_json_string = json.loads(user_list.content.decode('utf-8'))
self.assertEqual(user_list.status_code, 200)
self.assertEqual(u_json_string[0]['username'], "*****@*****.**")
self.assertEqual(u_json_string[1]['username'],
"*****@*****.**")
self.assertEqual(u_json_string[0]['role'], UserProfile.INSTRUCTOR)
self.assertEqual(u_json_string[1]['role'], UserProfile.STUDENT)
self.assertEqual(u_json_string[0]['first_name'], "Fa")
self.assertEqual(u_json_string[1]['first_name'], "Fa")
def test_get_user_detail(self):
user_detail1 = self.client.get('/users/%s/' % self.user1.pk)
u1_json_string = json.loads(user_detail1.content.decode('utf-8'))
user_detail2 = self.client.get('/users/%s/' % self.user2.pk)
u2_json_string = json.loads(user_detail2.content.decode('utf-8'))
self.assertEqual(user_detail1.status_code, 200)
self.assertEqual(user_detail2.status_code, 200)
self.assertEqual(u1_json_string['username'], "*****@*****.**")
self.assertEqual(u2_json_string['username'], "*****@*****.**")
self.assertEqual(u1_json_string['role'], UserProfile.INSTRUCTOR)
self.assertEqual(u2_json_string['role'], UserProfile.STUDENT)
self.assertEqual(u1_json_string['first_name'], "Fa")
self.assertEqual(u2_json_string['first_name'], "Fa")
def test_get_bad_user_detail(self):
user_detail = self.client.get('/users/9999/')
u_json_string = json.loads(user_detail.content.decode('utf-8'))
self.assertEqual(user_detail.status_code, 404)
self.assertEqual(u_json_string['detail'], "Not found.")
def test_full_put_user(self):
self.assertEqual(self.user1.userprofile.role, UserProfile.INSTRUCTOR)
self.assertEqual(self.user1.first_name, "Fa")
# Updated user information
new_info = {
"first_name": "Hello",
"username": "******",
"role": UserProfile.STUDENT,
}
user_put = self.client.put('/users/%s/' % self.user1.pk,
json.dumps(new_info),
content_type="application/json")
self.assertEqual(user_put.status_code, 200)
self.assertEqual(
User.objects.get(pk=self.user1.pk).userprofile.role,
UserProfile.STUDENT)
self.assertEqual(
User.objects.get(pk=self.user1.pk).first_name, "Hello")
def test_partial_put_user(self):
self.assertEqual(self.user1.userprofile.role, UserProfile.INSTRUCTOR)
self.assertEqual(self.user1.first_name, "Fa")
# Updated user information
new_info = {
"first_name": "Hello",
"username": "******",
}
user_put = self.client.put('/users/%s/' % self.user1.pk,
json.dumps(new_info),
content_type="application/json")
self.assertEqual(user_put.status_code, 200)
self.assertEqual(
User.objects.get(pk=self.user1.pk).userprofile.role,
UserProfile.INSTRUCTOR)
self.assertEqual(
User.objects.get(pk=self.user1.pk).first_name, "Hello")
def test_bad_put_user(self):
# Updated user information
new_info = {
"first_name": "Hello",
"username": "",
"role": UserProfile.STUDENT,
}
user_put = self.client.put('/users/%s/' % self.user1.pk,
json.dumps(new_info),
content_type="application/json")
self.assertEqual(user_put.status_code, 400)
def test_delete_user(self):
self.user3 = User(username="******",
first_name="Ryan",
password="******")
self.user3.save()
user_detail = self.client.get('/users/%s/' % self.user3.pk)
u_json_string = json.loads(user_detail.content.decode('utf-8'))
self.assertEqual(user_detail.status_code, 200)
user_delete = self.client.delete('/users/%s/' % self.user3.pk)
self.assertEqual(user_delete.status_code, 204)
user_detail = self.client.get('/users/%s/' % self.user3.pk)
u_json_string = json.loads(user_detail.content.decode('utf-8'))
self.assertEqual(user_detail.status_code, 404)
def add_user(request):
'''用户添加'''
context = {}
context['add_user'] = True
if request.method == 'GET':
#获得职位信息
positions = Position.objects.all()
position_list = []
for position in positions:
position_list.append({
'id': position.id,
'name': position.name,
})
#获得部门信息
departments = Department.objects.all()
department_list = []
for department in departments:
department_list.append({
'id': department.id,
'name': department.name,
})
if department_list:
context['department_list'] = department_list
if position_list:
context['position_list'] = position_list
else:
errorList = []
username = request.POST['username']
username_zh = request.POST.get('username_zh', '')
mobile = request.POST['mobile']
password = request.POST['password']
confirm_password = request.POST['confirm_password']
department = request.POST.get('department', '')
position = request.POST.get('position', '')
chief = request.POST.get('chief', '')
short_phone = request.POST.get('short_phone', '')
if password != confirm_password:
errorList.append({'error': u'密码不一致'})
else:
user = User()
if username:
user.username = username
if username_zh:
user.username_zh = username_zh
if mobile:
user.mobile = mobile
if department:
user.department = Department.objects.get(id=int(department))
if position:
user.position = Position.objects.get(id=int(position))
if short_phone:
user.short_phone = short_phone
user.set_password(password)
try:
user.full_clean()
user.save()
except ValidationError as e:
# Do something based on the errors contained in e.message_dict.
# Display them to a user, or handle them programmatically.
errKeys = e.message_dict.keys()
for errKey in errKeys:
errorList.append({'error': e.message_dict[errKey][0]})
if errorList:
context['errorList'] = errorList
return render(request, 'base/settings.html', context)
sb1.save(); sb2.save(); sb3.save(); t1 = Tag(tag_name='government schools'); t2 = Tag(tag_name='advertising'); t3 = Tag(tag_name='legal'); t4 = Tag(tag_name='community project'); t1.save(); t2.save(); t3.save(); t4.save(); u1 = User(user_email='*****@*****.**'); u1.save(); u2 = User(user_email='*****@*****.**'); u2.save(); sk1 = Skill(skill_name='Information Architecture'); sk2 = Skill(skill_name='Web Development'); sk3 = Skill(skill_name='Photography&Filmography'); sk4 = Skill(skill_name='Brand Management'); sk5 = Skill(skill_name='Content Generation'); sk6 = Skill(skill_name='Stratergy Consulting'); sk1.save(); sk2.save(); sk3.save(); sk4.save(); sk5.save();
