diff --git a/api/views.py b/api/views.py index f8cfd517a0..ff6514de92 100644 --- a/api/views.py +++ b/api/views.py @@ -1,13 +1,26 @@ -# Create your views here. +# encoding: utf-8 import re import sys import os import stat import simplejson as json +import settings + from django.http import HttpResponse, HttpResponseServerError +from django.contrib.sites.models import RequestSite +from django.core.urlresolvers import reverse +from django.views.decorators.csrf import csrf_exempt +from django.template import loader -from auth.decorators import login_required, api_login_required +from djangorestframework.renderers import JSONRenderer +from djangorestframework.compat import View +from djangorestframework.mixins import ResponseMixin +from djangorestframework.response import Response +from auth.decorators import api_login_required +from auth import login as auth_login + +from pysearpc import SearpcError from seaserv import ccnet_rpc, ccnet_threaded_rpc, get_repos, \ get_repo, get_commits, get_branches, \ seafserv_threaded_rpc, seafserv_rpc, get_binding_peerids, \ @@ -16,20 +29,12 @@ from seaserv import ccnet_rpc, ccnet_threaded_rpc, get_repos, \ from seahub.utils import list_to_string, \ get_httpserver_root, gen_token, \ check_filename_with_rename, get_accessible_repos, EMPTY_SHA1, \ - gen_file_get_url + gen_file_get_url, string2list from seahub.views import access_to_repo, validate_owner -from pysearpc import SearpcError +from seahub.contacts.signals import mail_sended -from djangorestframework.renderers import JSONRenderer -from djangorestframework.compat import View -from djangorestframework.mixins import ResponseMixin -from djangorestframework.response import Response -from django.core.urlresolvers import reverse - -from auth.forms import AuthenticationForm -from auth import login as auth_login -from django.views.decorators.csrf import csrf_exempt +from share.models import FileShare from mime import MIME_MAP @@ -107,7 +112,6 @@ def get_dir_entrys_by_id(request, dir_id): try: entry["size"] = get_file_size(dirent.obj_id) except Exception, e: - print e entry["size"]=0 entry["type"]=dtype @@ -298,18 +302,92 @@ class RepoDirIdView(ResponseMixin, View): return get_dir_entrys_by_id(request, dir_id) +def get_shared_link(request, repo_id, path): + l = FileShare.objects.filter(repo_id=repo_id).filter(\ + username=request.user.username).filter(path=path) + token = None + if len(l) > 0: + fileshare = l[0] + token = fileshare.token + else: + token = gen_token(max_length=10) -def get_repo_file(request, repo_id, file_id, file_name): - token = gen_token() - op='download' - seafserv_rpc.web_save_access_token(token, repo_id, file_id, - op, request.user.username) - redirect_url = gen_file_get_url(token, file_name) + fs = FileShare() + fs.username = request.user.username + fs.repo_id = repo_id + fs.path = path + fs.token = token - response = HttpResponse(json.dumps(redirect_url), status=200, - content_type=json_content_type) - response["oid"] = file_id - return response + try: + fs.save() + except IntegrityError, e: + err = '获取分享链接失败，请重新获取' + return api_err(request, '500', err) + + domain = RequestSite(request).domain + file_shared_link = 'http://%s%sf/%s/' % (domain, + settings.SITE_ROOT, token) + return HttpResponse(json.dumps(file_shared_link), status=200, content_type=json_content_type) + +def get_repo_file(request, repo_id, file_id, file_name, op): + if op == 'download': + token = gen_token() + seafserv_rpc.web_save_access_token(token, repo_id, file_id, + op, request.user.username) + redirect_url = gen_file_get_url(token, file_name) + response = HttpResponse(json.dumps(redirect_url), status=200, + content_type=json_content_type) + response["oid"] = file_id + return response + + if op == 'sharelink': + path = request.GET.get('p', None) + assert path, 'path must be passed in the url' + return get_shared_link(request, repo_id, path) + +def send_share_link(request, repo_id, path, emails): + l = FileShare.objects.filter(repo_id=repo_id).filter(\ + username=request.user.username).filter(path=path) + if len(l) > 0: + fileshare = l[0] + token = fileshare.token + else: + token = gen_token(max_length=10) + + fs = FileShare() + fs.username = request.user.username + fs.repo_id = repo_id + fs.path = path + fs.token = token + + try: + fs.save() + except IntegrityError, e: + err = '获取分享链接失败，请重新获取' + return api_err(request, '500', err) + + domain = RequestSite(request).domain + file_shared_link = 'http://%s%sf/%s/' % (domain, + settings.SITE_ROOT, token) + + t = loader.get_template('shared_link_email.html') + to_email_list = string2list(emails) + for to_email in to_email_list: + mail_sended.send(sender=None, user=request.user.username, + email=to_email) + c = { + 'email': request.user.username, + 'to_email': to_email, + 'file_shared_link': file_shared_link, + } + try: + send_mail('您的好友通过SeaCloud分享了一个文件给您', + t.render(Context(c)), None, [to_email], + fail_silently=False) + except: + err = 'Failed to send' + return api_error(request, '406', err) + return HttpResponse(json.dumps(file_shared_link), status=200, content_type=json_content_type) class RepoFileIdView(ResponseMixin, View): renderers = (JSONRenderer,) @@ -322,7 +400,7 @@ class RepoFileIdView(ResponseMixin, View): return resp file_name = request.GET.get('file_name', file_id) - return get_repo_file (request, repo_id, file_id, file_name) + return get_repo_file (request, repo_id, file_id, file_name, 'download') class RepoFilePathView(ResponseMixin, View): @@ -340,17 +418,37 @@ class RepoFilePathView(ResponseMixin, View): file_id = None try: - file_id = seafserv_threaded_rpc.get_file_by_path(commit.id, + file_id = seafserv_threaded_rpc.get_file_by_path(repo_id, path.encode('utf-8')) except SearpcError, e: - return api_error(request, '403', e.msg) + return api_error(request, '404', e.msg) if not file_id: return api_error(request, '400', "Path invalid") file_name = request.GET.get('file_name', file_id) - return get_repo_file(request, repo_id, file_name) + op = request.GET.get('op', 'download') + return get_repo_file(request, repo_id, file_id, file_name, op) @api_login_required def post(self, request, repo_id): - pass + repo = get_repo(repo_id) + resp = check_repo_access_permission(request, repo) + if resp: + return resp + + path = request.GET.get('p', None) + if not path: + return api_error(request, '404', "Path invalid") + + op = request.GET.get('op', 'delete') + if op == 'delete': + pass + if op == 'sendsharelink': + emails = request.POST.get('email', None) + if not emails: + return api_error(request, '400', 'emails required') + return send_share_link(request, path, emails) + + return api_error(request, '404', 'operation not support') + diff --git a/urls.py b/urls.py index e2ac848653..7d4dcde689 100644 --- a/urls.py +++ b/urls.py @@ -48,7 +48,7 @@ urlpatterns = patterns('', (r'^sharedlink/get/$', get_shared_link), (r'^sharedlink/remove/$', remove_shared_link), (r'^sharedlink/send/$', send_shared_link), - (r'^f/(?P[^/]+)/$', view_shared_file), + url(r'^f/(?P[^/]+)/$', view_shared_file, name='view_shared_file'), (r'^file_upload_progress_page/$', file_upload_progress_page), (r'^repo/new_dir/$', repo_new_dir),