mirror of
https://github.com/haiwen/seahub.git
synced 2025-09-21 03:18:23 +00:00
127 lines
4.0 KiB
Python
127 lines
4.0 KiB
Python
# Copyright (c) 2012-2016 Seafile Ltd.
|
|
from django.db import models
|
|
from django.urls import reverse
|
|
from django.utils import timezone
|
|
from django.utils.translation import gettext_lazy as _
|
|
from seaserv import seafile_api
|
|
|
|
from seahub.base.fields import LowerCaseCharField
|
|
from seahub.base.templatetags.seahub_tags import email2nickname
|
|
from seahub.avatar.templatetags.avatar_tags import api_avatar_url
|
|
from seahub.utils import get_site_scheme_and_netloc
|
|
from seahub.utils.timeutils import (timestamp_to_isoformat_timestr,
|
|
datetime_to_isoformat_timestr)
|
|
|
|
class WikiDoesNotExist(Exception):
|
|
pass
|
|
|
|
class WikiPageMissing(Exception):
|
|
pass
|
|
|
|
class DuplicateWikiNameError(Exception):
|
|
pass
|
|
|
|
|
|
class WikiManager(models.Manager):
|
|
def add(self, wiki_name, username, permission='private', repo_id=None,
|
|
org_id=-1):
|
|
if not permission:
|
|
permission = 'private'
|
|
|
|
from .utils import slugfy_wiki_name
|
|
slug = slugfy_wiki_name(wiki_name)
|
|
if self.filter(slug=slug).count() > 0:
|
|
raise DuplicateWikiNameError
|
|
|
|
now = timezone.now()
|
|
if repo_id is None: # create new repo to store the wiki pages
|
|
if org_id and org_id > 0:
|
|
repo_id = seafile_api.create_org_repo(wiki_name, '', username, org_id)
|
|
else:
|
|
repo_id = seafile_api.create_repo(wiki_name, '', username)
|
|
|
|
repo = seafile_api.get_repo(repo_id)
|
|
assert repo is not None
|
|
|
|
wiki = self.model(username=username, name=wiki_name, slug=slug,
|
|
repo_id=repo.id, permission=permission,
|
|
created_at=now)
|
|
wiki.save(using=self._db)
|
|
return wiki
|
|
|
|
|
|
class Wiki(models.Model):
|
|
"""New wiki model to enable a user has multiple wikis and replace
|
|
personal wiki.
|
|
"""
|
|
PERM_CHOICES = (
|
|
('private', 'private'),
|
|
('public', 'public'),
|
|
)
|
|
|
|
username = LowerCaseCharField(max_length=255)
|
|
name = models.CharField(max_length=255)
|
|
slug = models.CharField(max_length=255, unique=True)
|
|
repo_id = models.CharField(max_length=36, db_index=True)
|
|
permission = models.CharField(max_length=50) # private, public
|
|
created_at = models.DateTimeField(default=timezone.now, db_index=True)
|
|
objects = WikiManager()
|
|
|
|
class Meta:
|
|
unique_together = (('username', 'repo_id'),)
|
|
ordering = ["name"]
|
|
|
|
@property
|
|
def link(self):
|
|
return get_site_scheme_and_netloc().rstrip('/') + reverse('wiki:slug',
|
|
args=[self.slug])
|
|
|
|
@property
|
|
def updated_at(self):
|
|
assert len(self.repo_id) == 36
|
|
|
|
repo = seafile_api.get_repo(self.repo_id)
|
|
if not repo:
|
|
return ''
|
|
|
|
return repo.last_modify
|
|
|
|
def has_read_perm(self, request):
|
|
from seahub.views import check_folder_permission
|
|
if self.permission == 'public':
|
|
return True
|
|
else: # private
|
|
if not request.user.is_authenticated:
|
|
return False
|
|
repo_perm = check_folder_permission(request, self.repo_id, '/')
|
|
if not repo_perm:
|
|
return False
|
|
return True
|
|
|
|
def to_dict(self):
|
|
avatar_url, is_default, date_uploaded = api_avatar_url(self.username, int(32))
|
|
return {
|
|
'id': self.pk,
|
|
'owner': self.username,
|
|
'owner_nickname': email2nickname(self.username),
|
|
'owner_avatar_url': avatar_url,
|
|
'name': self.name,
|
|
'slug': self.slug,
|
|
'link': self.link,
|
|
'permission': self.permission,
|
|
'created_at': datetime_to_isoformat_timestr(self.created_at),
|
|
'updated_at': timestamp_to_isoformat_timestr(self.updated_at),
|
|
'repo_id': self.repo_id,
|
|
}
|
|
|
|
|
|
###### signal handlers
|
|
from django.dispatch import receiver
|
|
from seahub.signals import repo_deleted
|
|
|
|
@receiver(repo_deleted)
|
|
def remove_wiki(sender, **kwargs):
|
|
repo_id = kwargs['repo_id']
|
|
|
|
Wiki.objects.filter(repo_id=repo_id).delete()
|