def teardown(test):
tw_util.get_vc = test.globs['real_get_vc']
image_utils.get_img = test.globs['real_get_img']
# テストで保存されたtwiconを削除
relative_pathname = os.path.join('twicon', str(ci.spa_id))
absolute_pathname = (
image_utils.build_media_absolute_pathname(relative_pathname))
if os.path.exists(absolute_pathname):
os.remove(absolute_pathname)
def get_img_mock(url):
valid_urls = {ci.spa_icon_url}
if url in valid_urls:
relative_pathname = 'default_twicon'
absolute_pathname = (
image_utils.build_media_absolute_pathname(relative_pathname))
image = open(absolute_pathname)
return image.read()
else:
return None
def setUp(self):
self.client.login(username='******', password='******')
# Uploadするファイルの絶対パス
relative_pathname_src = 'default_twicon'
self.absolute_pathname_src = (
image_utils.build_media_absolute_pathname(relative_pathname_src))
# 存在確認
errmsg = 'File [%s] is not Found' % relative_pathname_src
assert os.path.exists(self.absolute_pathname_src), errmsg
# 保存後のファイルの絶対パス
relative_pathname = os.path.join('uploads', 'img_1')
self.absolute_pathname = (
image_utils.build_media_absolute_pathname(relative_pathname))
# テストなどで保存された画像ファイルがあれば削除
if os.path.exists(self.absolute_pathname):
os.remove(self.absolute_pathname)
def setUp(self):
self._original_get_vc = tw_util.get_vc
tw_util.get_vc = test_views.get_vc_mock
self._original_get_img = image_utils.get_img
image_utils.get_img = test_views.get_img_mock
# 既にtwiconが保存されていれば削除
relative_pathname = os.path.join('twicon', str(ci.spa_id))
self.absolute_pathname = (
image_utils.build_media_absolute_pathname(relative_pathname))
if os.path.exists(self.absolute_pathname):
os.remove(self.absolute_pathname)
def setup(test):
Post.objects.all().delete()
Lecture.objects.all().delete()
User.objects.all().delete()
test.globs['shortcuts'] = shortcuts
test.globs['Client'] = Client
test.globs['real_get_vc'] = tw_util.get_vc
tw_util.get_vc = get_vc_mock
test.globs['real_get_img'] = image_utils.get_img
image_utils.get_img = get_img_mock
# 既にtwiconが保存されていれば削除
relative_pathname = os.path.join('twicon', str(ci.spa_id))
absolute_pathname = (
image_utils.build_media_absolute_pathname(relative_pathname))
if os.path.exists(absolute_pathname):
os.remove(absolute_pathname)
def auth_view(request):
try:
key = request.GET['access_token_key']
secret = request.GET['access_token_secret']
except MultiValueDictKeyError:
# bad request
return json_response_bad_request()
try:
# get twitter account by key and secret
tw_account = tw_util.get_vc(key, secret)
except TypeError:
# Error reason is not well known
# sending dummy access token key/secret causes error
return json_response_server_error()
if tw_account == {}:
# 正しいアクセストークンキー、シークレットでなかった場合 など
return json_response_not_found()
user_name = tw_account['id']
# get twitter icon URL and save icon image to local
# 暫定的に認証時に毎回アイコンを取得
twicon = image_utils.get_img(tw_account['icon_url'])
if twicon is None:
relative_pathname = 'default_twicon'
else:
relative_pathname = os.path.join('twicon', str(user_name))
absolute_pathname = image_utils.build_media_absolute_pathname(
relative_pathname)
image_utils.save_bindata(absolute_pathname, twicon)
icon_url = image_utils.build_media_absolute_url(request, relative_pathname)
# 新規に作成されたユーザーも、登録済みだったユーザーも
# どちらもパスワードとしてtemp_passwordを設定する
temp_password = User.objects.make_random_password()
try:
# HQTP user exists
user = User.objects.get(username=user_name)
user.set_password(temp_password)
user.save()
created = False
except User.DoesNotExist:
# User has twitter account, but doesn't have HQTP account
# create new user
user = User.objects.create_user(username=user_name,
email='',
password=temp_password)
profile = user.get_profile()
profile.screen_name = tw_account['screen_name']
profile.name = tw_account['name']
profile.icon_url = icon_url
profile.save()
achieve.give_achievement('first_login', user)
created = True
auth_user = authenticate(username=user_name, password=temp_password)
# セキュリティのために、パスワード認証ができないようにします
# アクセストークンKEY、SECRETによる認証しか行いません
auth_user.set_unusable_password()
if auth_user.is_active:
# Log in successful
login(request, auth_user)
user_info = user_to_dict(auth_user)
else:
# User is deleted
return json_response_not_found()
return json_response(context=dict(created=created, user=user_info))
def lecture_timeline_view(request):
if request.method == 'GET':
try:
# get timeline
lecture_id = int(request.GET['id'])
except (MultiValueDictKeyError, ValueError):
# key 'id' is not requested
# or id is not integer
return json_response_bad_request()
if 'since_id' in request.GET:
try:
since_id = int(request.GET['since_id'])
except ValueError:
return json_response_bad_request()
else:
since_id = 0
if not request.user.is_authenticated():
# get need authentication
return json_response_forbidden()
try:
lec = Lecture.objects.get(pk=lecture_id)
except Lecture.DoesNotExist:
# invalid lecture ID
return json_response_not_found()
if achieve.first_or_interval(request.user):
achieve.give_achievement('attend_lecture', request.user)
# successfully get timeline
posts = [q.to_dict() for q in
lec.post_set.filter(pk__gt=since_id).order_by('virtual_ts')]
return json_response(context=dict(posts=posts))
elif request.method == 'POST':
try:
lecture_id = int(request.POST['id'])
except (MultiValueDictKeyError, ValueError):
return json_response_bad_request()
# boolean flags
use_before_vts = ('before_virtual_ts' in request.POST)
use_after_vts = ('after_virtual_ts' in request.POST)
use_text = ('body' in request.POST)
use_image = ('image' in request.FILES)
if use_before_vts != use_after_vts:
# only one is requested and the other one is not
# NOTE: != is logical exclusive-or
return json_response_bad_request()
if use_text == use_image:
# bodyとimageのどちらか一方を指定
return json_response_bad_request()
if use_before_vts and use_after_vts:
# post to between 2 posts
try:
before_vts = int(request.POST['before_virtual_ts'])
after_vts = int(request.POST['after_virtual_ts'])
except ValueError:
return json_response_bad_request()
vts = Post.calc_mid(before_vts, after_vts)
else:
# post to latest
vts = Post.time_to_vts(time())
if not request.user.is_authenticated():
# get need authentication
return json_response_forbidden()
try:
lec = Lecture.objects.get(pk=lecture_id)
except Lecture.DoesNotExist:
# invalid lecture ID
return json_response_not_found()
post = lec.post_set.create(added_by=request.user,
virtual_ts=vts)
if use_text:
post.body = request.POST['body']
post.save()
if achieve.contains_url(request.POST['body']):
achieve.give_achievement('upload_url', request.user)
if achieve.contains_specialwords(request.POST['body']):
achieve.give_achievement('easter_egg', request.user)
elif use_image:
# save image file
# ユニークなfilenameとして、Postのpkを使う
image = request.FILES['image']
filename = 'img_' + str(post.pk)
relative_pathname = os.path.join('uploads', filename)
absolute_pathname = image_utils.build_media_absolute_pathname(
relative_pathname)
image_utils.save_bindata(absolute_pathname, image.read())
image_url = image_utils.build_media_absolute_url(
request, relative_pathname)
post.image_url = image_url
post.save()
achieve.give_achievement('upload_image', request.user)
if achieve.is_post_5_minutes(lec, request.user):
achieve.give_achievement('consecutive_post', request.user)
achieve.give_achievement('one_post', request.user)
if use_before_vts:
for p in Post.objects.filter(
virtual_ts=before_vts).exclude(added_by=request.user):
achieve.give_achievement('post_inserted', p.added_by)
return json_response(context=dict(post=post.to_dict()))