12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475 |
- # coding: utf8
- import logging
-
- import os
-
- import transaction
- import typing
- import re
- from datetime import datetime
- from time import mktime
- from os.path import dirname, basename
-
- from sqlalchemy.orm import Session
-
- from tracim.config import CFG
- from tracim.lib.core.content import ContentApi
- from tracim.lib.core.user import UserApi
- from tracim.lib.webdav.utils import transform_to_display, HistoryType, \
- FakeFileStream
- from tracim.lib.webdav.utils import transform_to_bdd
- from tracim.lib.core.workspace import WorkspaceApi
- from tracim.models.data import User, ContentRevisionRO
- from tracim.models.data import Workspace
- from tracim.models.data import Content, ActionDescription
- from tracim.models.data import ContentType
- from tracim.lib.webdav.design import designThread, designPage
-
- from wsgidav import compat
- from wsgidav.dav_error import DAVError, HTTP_FORBIDDEN
- from wsgidav.dav_provider import DAVCollection, DAVNonCollection
- from wsgidav.dav_provider import _DAVResource
- from tracim.lib.webdav.utils import normpath
-
- from sqlalchemy.orm.exc import NoResultFound, MultipleResultsFound
-
- from tracim.models.revision_protection import new_revision
-
- logger = logging.getLogger()
-
-
- class ManageActions(object):
- """
- This object is used to encapsulate all Deletion/Archiving related
- method as to not duplicate too much code
- """
- def __init__(self,
- session: Session,
- action_type: str,
- api: ContentApi,
- content: Content
- ):
- self.session = session
- self.content_api = api
- self.content = content
-
- self._actions = {
- ActionDescription.ARCHIVING: self.content_api.archive,
- ActionDescription.DELETION: self.content_api.delete,
- ActionDescription.UNARCHIVING: self.content_api.unarchive,
- ActionDescription.UNDELETION: self.content_api.undelete
- }
-
- self._type = action_type
-
- def action(self):
- with new_revision(
- session=self.session,
- tm=transaction.manager,
- content=self.content,
- ):
- self._actions[self._type](self.content)
- self.content_api.save(self.content, self._type)
-
- transaction.commit()
-
-
- class RootResource(DAVCollection):
- """
- RootResource ressource that represents tracim's home, which contains all workspaces
- """
-
- def __init__(self, path: str, environ: dict, user: User, session: Session):
- super(RootResource, self).__init__(path, environ)
-
- self.user = user
- self.session = session
- # TODO BS 20170221: Web interface should list all workspace to. We
- # disable it here for moment. When web interface will be updated to
- # list all workspace, change this here to.
- self.workspace_api = WorkspaceApi(
- current_user=self.user,
- session=session,
- force_role=True
- )
-
- def __repr__(self) -> str:
- return '<DAVCollection: RootResource>'
-
- def getMemberNames(self) -> [str]:
- """
- This method returns the names (here workspace's labels) of all its children
-
- Though for perfomance issue, we're not using this function anymore
- """
- return [workspace.label for workspace in self.workspace_api.get_all()]
-
- def getMember(self, label: str) -> DAVCollection:
- """
- This method returns the child Workspace that corresponds to a given name
-
- Though for perfomance issue, we're not using this function anymore
- """
- try:
- workspace = self.workspace_api.get_one_by_label(label)
- workspace_path = '%s%s%s' % (self.path, '' if self.path == '/' else '/', transform_to_display(workspace.label))
-
- return WorkspaceResource(
- workspace_path,
- self.environ,
- workspace,
- session=self.session,
- user=self.user,
- )
- except AttributeError:
- return None
-
- def createEmptyResource(self, name: str):
- """
- This method is called whenever the user wants to create a DAVNonCollection resource (files in our case).
-
- There we don't allow to create files at the root;
- only workspaces (thus collection) can be created.
- """
- raise DAVError(HTTP_FORBIDDEN)
-
- def createCollection(self, name: str):
- """
- This method is called whenever the user wants to create a DAVCollection resource as a child (in our case,
- we create workspaces as this is the root).
-
- [For now] we don't allow to create new workspaces through
- webdav client. Though if we come to allow it, deleting the error's raise will
- make it possible.
- """
- # TODO : remove comment here
- # raise DAVError(HTTP_FORBIDDEN)
-
- new_workspace = self.workspace_api.create_workspace(name)
- self.workspace_api.save(new_workspace)
-
- workspace_path = '%s%s%s' % (
- self.path, '' if self.path == '/' else '/', transform_to_display(new_workspace.label))
-
- transaction.commit()
- return WorkspaceResource(
- workspace_path,
- self.environ,
- new_workspace,
- user=self.user,
- session=self.session,
- )
-
- def getMemberList(self):
- """
- This method is called by wsgidav when requesting with a depth > 0, it will return a list of _DAVResource
- of all its direct children
- """
-
- members = []
- for workspace in self.workspace_api.get_all():
- workspace_path = '%s%s%s' % (self.path, '' if self.path == '/' else '/', workspace.label)
- members.append(
- WorkspaceResource(
- path=workspace_path,
- environ=self.environ,
- workspace=workspace,
- user=self.user,
- session=self.session,
- )
- )
-
- return members
-
-
- class WorkspaceResource(DAVCollection):
- """
- Workspace resource corresponding to tracim's workspaces.
- Direct children can only be folders, though files might come later on and are supported
- """
-
- def __init__(self,
- path: str,
- environ: dict,
- workspace: Workspace,
- user: User,
- session: Session
- ) -> None:
- super(WorkspaceResource, self).__init__(path, environ)
-
- self.workspace = workspace
- self.content = None
- self.user = user
- self.session = session
- self.content_api = ContentApi(
- current_user=self.user,
- session=session,
- config=self.provider.app_config,
- show_temporary=True
- )
-
- self._file_count = 0
-
- def __repr__(self) -> str:
- return "<DAVCollection: Workspace (%d)>" % self.workspace.workspace_id
-
- def getPreferredPath(self):
- return self.path
-
- def getCreationDate(self) -> float:
- return mktime(self.workspace.created.timetuple())
-
- def getDisplayName(self) -> str:
- return self.workspace.label
-
- def getLastModified(self) -> float:
- return mktime(self.workspace.updated.timetuple())
-
- def getMemberNames(self) -> [str]:
- retlist = []
-
- children = self.content_api.get_all(
- parent_id=self.content.id if self.content is not None else None,
- workspace=self.workspace
- )
-
- for content in children:
- # the purpose is to display .history only if there's at least one content's type that has a history
- if content.type != ContentType.Folder:
- self._file_count += 1
- retlist.append(content.get_label_as_file())
-
- return retlist
-
- def getMember(self, content_label: str) -> _DAVResource:
-
- return self.provider.getResourceInst(
- '%s/%s' % (self.path, transform_to_display(content_label)),
- self.environ
- )
-
- def createEmptyResource(self, file_name: str):
- """
- [For now] we don't allow to create files right under workspaces.
- Though if we come to allow it, deleting the error's raise will make it possible.
- """
- # TODO : remove commentary here raise DAVError(HTTP_FORBIDDEN)
- if '/.deleted/' in self.path or '/.archived/' in self.path:
- raise DAVError(HTTP_FORBIDDEN)
-
- content = None
-
- # Note: To prevent bugs, check here again if resource already exist
- path = os.path.join(self.path, file_name)
- resource = self.provider.getResourceInst(path, self.environ)
- if resource:
- content = resource.content
-
- return FakeFileStream(
- session=self.session,
- file_name=file_name,
- content_api=self.content_api,
- workspace=self.workspace,
- content=content,
- parent=self.content,
- path=self.path + '/' + file_name
- )
-
- def createCollection(self, label: str) -> 'FolderResource':
- """
- Create a new folder for the current workspace. As it's not possible for the user to choose
- which types of content are allowed in this folder, we allow allow all of them.
-
- This method return the DAVCollection created.
- """
-
- if '/.deleted/' in self.path or '/.archived/' in self.path:
- raise DAVError(HTTP_FORBIDDEN)
-
- folder = self.content_api.create(
- content_type=ContentType.Folder,
- workspace=self.workspace,
- label=label,
- parent=self.content
- )
-
- subcontent = dict(
- folder=True,
- thread=True,
- file=True,
- page=True
- )
-
- self.content_api.set_allowed_content(folder, subcontent)
- self.content_api.save(folder)
-
- transaction.commit()
-
- return FolderResource('%s/%s' % (self.path, transform_to_display(label)),
- self.environ,
- content=folder,
- session=self.session,
- user=self.user,
- workspace=self.workspace,
- )
-
- def delete(self):
- """For now, it is not possible to delete a workspace through the webdav client."""
- raise DAVError(HTTP_FORBIDDEN)
-
- def supportRecursiveMove(self, destpath):
- return True
-
- def moveRecursive(self, destpath):
- if dirname(normpath(destpath)) == self.environ['http_authenticator.realm']:
- self.workspace.label = basename(normpath(destpath))
- transaction.commit()
- else:
- raise DAVError(HTTP_FORBIDDEN)
-
- def getMemberList(self) -> [_DAVResource]:
- members = []
-
- children = self.content_api.get_all(False, ContentType.Any, self.workspace)
-
- for content in children:
- content_path = '%s/%s' % (self.path, transform_to_display(content.get_label_as_file()))
-
- if content.type == ContentType.Folder:
- members.append(
- FolderResource(
- path=content_path,
- environ=self.environ,
- workspace=self.workspace,
- user=self.user,
- content=content,
- session=self.session,
- )
- )
- elif content.type == ContentType.File:
- self._file_count += 1
- members.append(
- FileResource(
- path=content_path,
- environ=self.environ,
- content=content,
- user=self.user,
- session=self.session,
- )
- )
- else:
- self._file_count += 1
- members.append(
- OtherFileResource(
- content_path,
- self.environ,
- content,
- session=self.session,
- user=self.user,
- ))
-
- if self._file_count > 0 and self.provider.show_history():
- members.append(
- HistoryFolderResource(
- path=self.path + '/' + ".history",
- environ=self.environ,
- content=self.content,
- workspace=self.workspace,
- type=HistoryType.Standard,
- session=self.session,
- user=self.user,
- )
- )
-
- if self.provider.show_delete():
- members.append(
- DeletedFolderResource(
- path=self.path + '/' + ".deleted",
- environ=self.environ,
- content=self.content,
- workspace=self.workspace,
- session=self.session,
- user=self.user,
- )
- )
-
- if self.provider.show_archive():
- members.append(
- ArchivedFolderResource(
- path=self.path + '/' + ".archived",
- environ=self.environ,
- content=self.content,
- workspace=self.workspace,
- user=self.user,
- session=self.session,
- )
- )
-
- return members
-
-
- class FolderResource(WorkspaceResource):
- """
- FolderResource resource corresponding to tracim's folders.
- Direct children can only be either folder, files, pages or threads
- By default when creating new folders, we allow them to contain all types of content
- """
-
- def __init__(
- self,
- path: str,
- environ: dict,
- workspace: Workspace,
- content: Content,
- user: User,
- session: Session
- ):
- super(FolderResource, self).__init__(
- path=path,
- environ=environ,
- workspace=workspace,
- user=user,
- session=session,
- )
- self.content = content
-
- def __repr__(self) -> str:
- return "<DAVCollection: Folder (%s)>" % self.content.label
-
- def getCreationDate(self) -> float:
- return mktime(self.content.created.timetuple())
-
- def getDisplayName(self) -> str:
- return transform_to_display(self.content.get_label_as_file())
-
- def getLastModified(self) -> float:
- return mktime(self.content.updated.timetuple())
-
- def delete(self):
- ManageActions(
- action_type=ActionDescription.DELETION,
- api=self.content_api,
- content=self.content,
- session=self.session,
- ).action()
-
- def supportRecursiveMove(self, destpath: str):
- return True
-
- def moveRecursive(self, destpath: str):
- """
- As we support recursive move, copymovesingle won't be called, though with copy it'll be called
- but i have to check if the client ever call that function...
- """
- destpath = normpath(destpath)
-
- invalid_path = False
-
- # if content is either deleted or archived, we'll check that we try moving it to the parent
- # if yes, then we'll unarchive / undelete them, else the action's not allowed
- if self.content.is_deleted or self.content.is_archived:
- # we remove all archived and deleted from the path and we check to the destpath
- # has to be equal or else path not valid
- # ex: /a/b/.deleted/resource, to be valid destpath has to be = /a/b/resource (no other solution)
- current_path = re.sub(r'/\.(deleted|archived)', '', self.path)
-
- if current_path == destpath:
- ManageActions(
- action_type=ActionDescription.UNDELETION if self.content.is_deleted else ActionDescription.UNARCHIVING,
- api=self.content_api,
- content=self.content,
- session=self.session,
- ).action()
- else:
- invalid_path = True
- # if the content is not deleted / archived, check if we're trying to delete / archive it by
- # moving it to a .deleted / .archived folder
- elif basename(dirname(destpath)) in ['.deleted', '.archived']:
- # same test as above ^
- dest_path = re.sub(r'/\.(deleted|archived)', '', destpath)
-
- if dest_path == self.path:
- ManageActions(
- action_type=ActionDescription.DELETION if '.deleted' in destpath else ActionDescription.ARCHIVING,
- api=self.content_api,
- content=self.content,
- session=self.session,
- ).action()
- else:
- invalid_path = True
- # else we check if the path is good (not at the root path / not in a deleted/archived path)
- # and we move the content
- else:
- invalid_path = any(x in destpath for x in ['.deleted', '.archived'])
- invalid_path = invalid_path or any(x in self.path for x in ['.deleted', '.archived'])
- invalid_path = invalid_path or dirname(destpath) == self.environ['http_authenticator.realm']
-
- if not invalid_path:
- self.move_folder(destpath)
-
- if invalid_path:
- raise DAVError(HTTP_FORBIDDEN)
-
- def move_folder(self, destpath):
-
- workspace_api = WorkspaceApi(
- current_user=self.user,
- session=self.session,
- )
- workspace = self.provider.get_workspace_from_path(
- normpath(destpath), workspace_api
- )
-
- parent = self.provider.get_parent_from_path(
- normpath(destpath),
- self.content_api,
- workspace
- )
-
- with new_revision(
- content=self.content,
- tm=transaction.manager,
- session=self.session,
- ):
- if basename(destpath) != self.getDisplayName():
- self.content_api.update_content(self.content, transform_to_bdd(basename(destpath)))
- self.content_api.save(self.content)
- else:
- if workspace.workspace_id == self.content.workspace.workspace_id:
- self.content_api.move(self.content, parent)
- else:
- self.content_api.move_recursively(self.content, parent, workspace)
-
- transaction.commit()
-
- def getMemberList(self) -> [_DAVResource]:
- members = []
- content_api = ContentApi(
- current_user=self.user,
- config=self.provider.app_config,
- session=self.session,
- )
- visible_children = content_api.get_all(
- self.content.content_id,
- ContentType.Any,
- self.workspace,
- )
-
- for content in visible_children:
- content_path = '%s/%s' % (self.path, transform_to_display(content.get_label_as_file()))
-
- try:
- if content.type == ContentType.Folder:
- members.append(
- FolderResource(
- path=content_path,
- environ=self.environ,
- workspace=self.workspace,
- content=content,
- user=self.user,
- session=self.session,
- )
- )
- elif content.type == ContentType.File:
- self._file_count += 1
- members.append(
- FileResource(
- path=content_path,
- environ=self.environ,
- content=content,
- user=self.user,
- session=self.session,
- ))
- else:
- self._file_count += 1
- members.append(
- OtherFileResource(
- path=content_path,
- environ=self.environ,
- content=content,
- user=self.user,
- session=self.session,
- ))
- except Exception as exc:
- logger.exception(
- 'Unable to construct member {}'.format(
- content_path,
- ),
- exc_info=True,
- )
-
- if self._file_count > 0 and self.provider.show_history():
- members.append(
- HistoryFolderResource(
- path=self.path + '/' + ".history",
- environ=self.environ,
- content=self.content,
- workspace=self.workspace,
- type=HistoryType.Standard,
- user=self.user,
- session=self.session,
- )
- )
-
- if self.provider.show_delete():
- members.append(
- DeletedFolderResource(
- path=self.path + '/' + ".deleted",
- environ=self.environ,
- content=self.content,
- workspace=self.workspace,
- user=self.user,
- session=self.session,
- )
- )
-
- if self.provider.show_archive():
- members.append(
- ArchivedFolderResource(
- path=self.path + '/' + ".archived",
- environ=self.environ,
- content=self.content,
- workspace=self.workspace,
- user=self.user,
- session=self.session,
- )
- )
-
- return members
-
- # TODO - G.M - 02-05-2018 - Check these object (History/Deleted/Archived Folder)
- # Those object are now not in used by tracim and also not tested,
-
-
- class HistoryFolderResource(FolderResource):
- """
- A virtual resource which contains a sub-folder for every files (DAVNonCollection) contained in the parent
- folder
- """
-
- def __init__(self,
- path,
- environ,
- workspace: Workspace,
- user: User,
- session: Session,
- content: Content=None,
- type: str=HistoryType.Standard
- ) -> None:
- super(HistoryFolderResource, self).__init__(
- path=path,
- environ=environ,
- workspace=workspace,
- content=content,
- user=user,
- session=session,
- )
-
- self._is_archived = type == HistoryType.Archived
- self._is_deleted = type == HistoryType.Deleted
-
- self.content_api = ContentApi(
- current_user=self.user,
- show_archived=self._is_archived,
- show_deleted=self._is_deleted,
- session=self.session,
- config=self.provider.app_config,
- )
-
- def __repr__(self) -> str:
- return "<DAVCollection: HistoryFolderResource (%s)>" % self.content.file_name
-
- def getCreationDate(self) -> float:
- return mktime(datetime.now().timetuple())
-
- def getDisplayName(self) -> str:
- return '.history'
-
- def getLastModified(self) -> float:
- return mktime(datetime.now().timetuple())
-
- def getMember(self, content_label: str) -> _DAVResource:
- content = self.content_api.get_one_by_label_and_parent(
- content_label=content_label,
- content_parent=self.content
- )
-
- return HistoryFileFolderResource(
- path='%s/%s' % (self.path, content.get_label_as_file()),
- environ=self.environ,
- content=content,
- session=self.session,
- user=self.user,
- )
-
- def getMemberNames(self) -> [str]:
- ret = []
-
- content_id = None if self.content is None else self.content.id
- for content in self.content_api.get_all(content_id, ContentType.Any, self.workspace):
- if (self._is_archived and content.is_archived or
- self._is_deleted and content.is_deleted or
- not (content.is_archived or self._is_archived or content.is_deleted or self._is_deleted))\
- and content.type != ContentType.Folder:
- ret.append(content.get_label_as_file())
-
- return ret
-
- def createEmptyResource(self, name: str):
- raise DAVError(HTTP_FORBIDDEN)
-
- def createCollection(self, name: str):
- raise DAVError(HTTP_FORBIDDEN)
-
- def delete(self):
- raise DAVError(HTTP_FORBIDDEN)
-
- def handleDelete(self):
- return True
-
- def handleCopy(self, destPath: str, depthInfinity):
- return True
-
- def handleMove(self, destPath: str):
- return True
-
- def getMemberList(self) -> [_DAVResource]:
- members = []
-
- if self.content:
- children = self.content.children
- else:
- children = self.content_api.get_all(False, ContentType.Any, self.workspace)
-
- for content in children:
- if content.is_archived == self._is_archived and content.is_deleted == self._is_deleted:
- members.append(HistoryFileFolderResource(
- path='%s/%s' % (self.path, content.get_label_as_file()),
- environ=self.environ,
- content=content,
- user=self.user,
- session=self.session,
- ))
-
- return members
-
-
- class DeletedFolderResource(HistoryFolderResource):
- """
- A virtual resources which exists for every folder or workspaces which contains their deleted children
- """
-
- def __init__(
- self,
- path: str,
- environ: dict,
- workspace: Workspace,
- user: User,
- session: Session,
- content: Content=None
- ):
- super(DeletedFolderResource, self).__init__(
- path=path,
- environ=environ,
- workspace=workspace,
- user=user,
- content=content,
- session=session,
- type=HistoryType.Deleted
- )
-
- self._file_count = 0
-
- def __repr__(self):
- return "<DAVCollection: DeletedFolderResource (%s)" % self.content.file_name
-
- def getCreationDate(self) -> float:
- return mktime(datetime.now().timetuple())
-
- def getDisplayName(self) -> str:
- return '.deleted'
-
- def getLastModified(self) -> float:
- return mktime(datetime.now().timetuple())
-
- def getMember(self, content_label) -> _DAVResource:
-
- content = self.content_api.get_one_by_label_and_parent(
- content_label=content_label,
- content_parent=self.content
- )
-
- return self.provider.getResourceInst(
- path='%s/%s' % (self.path, transform_to_display(content.get_label_as_file())),
- environ=self.environ
- )
-
- def getMemberNames(self) -> [str]:
- retlist = []
-
- if self.content:
- children = self.content.children
- else:
- children = self.content_api.get_all(False, ContentType.Any, self.workspace)
-
- for content in children:
- if content.is_deleted:
- retlist.append(content.get_label_as_file())
-
- if content.type != ContentType.Folder:
- self._file_count += 1
-
- return retlist
-
- def getMemberList(self) -> [_DAVResource]:
- members = []
-
- if self.content:
- children = self.content.children
- else:
- children = self.content_api.get_all(False, ContentType.Any, self.workspace)
-
- for content in children:
- if content.is_deleted:
- content_path = '%s/%s' % (self.path, transform_to_display(content.get_label_as_file()))
-
- if content.type == ContentType.Folder:
- members.append(
- FolderResource(
- content_path,
- self.environ,
- self.workspace,
- content,
- user=self.user,
- session=self.session,
- ))
- elif content.type == ContentType.File:
- self._file_count += 1
- members.append(
- FileResource(
- content_path,
- self.environ,
- content,
- user=self.user,
- session=self.session,
- )
- )
- else:
- self._file_count += 1
- members.append(
- OtherFileResource(
- content_path,
- self.environ,
- content,
- user=self.user,
- session=self.session,
- ))
-
- if self._file_count > 0 and self.provider.show_history():
- members.append(
- HistoryFolderResource(
- path=self.path + '/' + ".history",
- environ=self.environ,
- content=self.content,
- workspace=self.workspace,
- user=self.user,
- type=HistoryType.Standard,
- session=self.session,
- )
- )
-
- return members
-
-
- class ArchivedFolderResource(HistoryFolderResource):
- """
- A virtual resources which exists for every folder or workspaces which contains their archived children
- """
- def __init__(
- self,
- path: str,
- environ: dict,
- workspace: Workspace,
- user: User,
- session: Session,
- content: Content=None
- ):
- super(ArchivedFolderResource, self).__init__(
- path=path,
- environ=environ,
- workspace=workspace,
- user=user,
- content=content,
- session=session,
- type=HistoryType.Archived
- )
-
- self._file_count = 0
-
- def __repr__(self) -> str:
- return "<DAVCollection: ArchivedFolderResource (%s)" % self.content.file_name
-
- def getCreationDate(self) -> float:
- return mktime(datetime.now().timetuple())
-
- def getDisplayName(self) -> str:
- return '.archived'
-
- def getLastModified(self) -> float:
- return mktime(datetime.now().timetuple())
-
- def getMember(self, content_label) -> _DAVResource:
-
- content = self.content_api.get_one_by_label_and_parent(
- content_label=content_label,
- content_parent=self.content
- )
-
- return self.provider.getResourceInst(
- path=self.path + '/' + transform_to_display(content.get_label_as_file()),
- environ=self.environ
- )
-
- def getMemberNames(self) -> [str]:
- retlist = []
-
- for content in self.content_api.get_all_with_filter(
- self.content if self.content is None else self.content.id, ContentType.Any):
- retlist.append(content.get_label_as_file())
-
- if content.type != ContentType.Folder:
- self._file_count += 1
-
- return retlist
-
- def getMemberList(self) -> [_DAVResource]:
- members = []
-
- if self.content:
- children = self.content.children
- else:
- children = self.content_api.get_all(False, ContentType.Any, self.workspace)
-
- for content in children:
- if content.is_archived:
- content_path = '%s/%s' % (self.path, transform_to_display(content.get_label_as_file()))
-
- if content.type == ContentType.Folder:
- members.append(
- FolderResource(
- content_path,
- self.environ,
- self.workspace,
- content,
- user=self.user,
- session=self.session,
- ))
- elif content.type == ContentType.File:
- self._file_count += 1
- members.append(
- FileResource(
- content_path,
- self.environ,
- content,
- user=self.user,
- session=self.session,
- ))
- else:
- self._file_count += 1
- members.append(
- OtherFileResource(
- content_path,
- self.environ,
- content,
- user=self.user,
- session=self.session,
- ))
-
- if self._file_count > 0 and self.provider.show_history():
- members.append(
- HistoryFolderResource(
- path=self.path + '/' + ".history",
- environ=self.environ,
- content=self.content,
- workspace=self.workspace,
- user=self.user,
- type=HistoryType.Standard,
- session=self.session,
- )
- )
-
- return members
-
-
- class HistoryFileFolderResource(HistoryFolderResource):
- """
- A virtual resource that contains for a given content (file/page/thread) all its revisions
- """
-
- def __init__(
- self,
- path: str,
- environ: dict,
- content: Content,
- user: User,
- session: Session
- ) -> None:
- super(HistoryFileFolderResource, self).__init__(
- path=path,
- environ=environ,
- workspace=content.workspace,
- content=content,
- user=user,
- session=session,
- type=HistoryType.All,
- )
-
- def __repr__(self) -> str:
- return "<DAVCollection: HistoryFileFolderResource (%s)" % self.content.file_name
-
- def getDisplayName(self) -> str:
- return self.content.get_label_as_file()
-
- def createCollection(self, name):
- raise DAVError(HTTP_FORBIDDEN)
-
- def getMemberNames(self) -> [int]:
- """
- Usually we would return a string, but here as we're working with different
- revisions of the same content, we'll work with revision_id
- """
- ret = []
-
- for content in self.content.revisions:
- ret.append(content.revision_id)
-
- return ret
-
- def getMember(self, item_id) -> DAVNonCollection:
-
- revision = self.content_api.get_one_revision(item_id)
-
- left_side = '%s/(%d - %s) ' % (self.path, revision.revision_id, revision.revision_type)
-
- if self.content.type == ContentType.File:
- return HistoryFileResource(
- path='%s%s' % (left_side, transform_to_display(revision.file_name)),
- environ=self.environ,
- content=self.content,
- content_revision=revision,
- session=self.session,
- user=self.user,
- )
- else:
- return HistoryOtherFile(
- path='%s%s' % (left_side, transform_to_display(revision.get_label_as_file())),
- environ=self.environ,
- content=self.content,
- content_revision=revision,
- session=self.session,
- user=self.user,
- )
-
- def getMemberList(self) -> [_DAVResource]:
- members = []
-
- for content in self.content.revisions:
-
- left_side = '%s/(%d - %s) ' % (self.path, content.revision_id, content.revision_type)
-
- if self.content.type == ContentType.File:
- members.append(HistoryFileResource(
- path='%s%s' % (left_side, transform_to_display(content.file_name)),
- environ=self.environ,
- content=self.content,
- content_revision=content,
- user=self.user,
- session=self.session,
- )
- )
- else:
- members.append(HistoryOtherFile(
- path='%s%s' % (left_side, transform_to_display(content.file_name)),
- environ=self.environ,
- content=self.content,
- content_revision=content,
- user=self.user,
- session=self.session,
- )
- )
-
- return members
-
-
- class FileResource(DAVNonCollection):
- """
- FileResource resource corresponding to tracim's files
- """
- def __init__(
- self,
- path: str,
- environ: dict,
- content: Content,
- user: User,
- session: Session,
- ) -> None:
- super(FileResource, self).__init__(path, environ)
-
- self.content = content
- self.user = user
- self.session = session
- self.content_api = ContentApi(
- current_user=self.user,
- config=self.provider.app_config,
- session=self.session,
- )
-
- # this is the property that windows client except to check if the file is read-write or read-only,
- # but i wasn't able to set this property so you'll have to look into it >.>
- # self.setPropertyValue('Win32FileAttributes', '00000021')
-
- def __repr__(self) -> str:
- return "<DAVNonCollection: FileResource (%d)>" % self.content.revision_id
-
- def getContentLength(self) -> int:
- return self.content.depot_file.file.content_length
-
- def getContentType(self) -> str:
- return self.content.file_mimetype
-
- def getCreationDate(self) -> float:
- return mktime(self.content.created.timetuple())
-
- def getDisplayName(self) -> str:
- return self.content.file_name
-
- def getLastModified(self) -> float:
- return mktime(self.content.updated.timetuple())
-
- def getContent(self) -> typing.BinaryIO:
- filestream = compat.BytesIO()
- filestream.write(self.content.depot_file.file.read())
- filestream.seek(0)
-
- return filestream
-
- def beginWrite(self, contentType: str=None) -> FakeFileStream:
- return FakeFileStream(
- content=self.content,
- content_api=self.content_api,
- file_name=self.content.get_label_as_file(),
- workspace=self.content.workspace,
- path=self.path,
- session=self.session,
- )
-
- def moveRecursive(self, destpath):
- """As we support recursive move, copymovesingle won't be called, though with copy it'll be called
- but i have to check if the client ever call that function..."""
- destpath = normpath(destpath)
-
- invalid_path = False
-
- # if content is either deleted or archived, we'll check that we try moving it to the parent
- # if yes, then we'll unarchive / undelete them, else the action's not allowed
- if self.content.is_deleted or self.content.is_archived:
- # we remove all archived and deleted from the path and we check to the destpath
- # has to be equal or else path not valid
- # ex: /a/b/.deleted/resource, to be valid destpath has to be = /a/b/resource (no other solution)
- current_path = re.sub(r'/\.(deleted|archived)', '', self.path)
-
- if current_path == destpath:
- ManageActions(
- action_type=ActionDescription.UNDELETION if self.content.is_deleted else ActionDescription.UNARCHIVING,
- api=self.content_api,
- content=self.content,
- session=self.session,
- ).action()
- else:
- invalid_path = True
- # if the content is not deleted / archived, check if we're trying to delete / archive it by
- # moving it to a .deleted / .archived folder
- elif basename(dirname(destpath)) in ['.deleted', '.archived']:
- # same test as above ^
- dest_path = re.sub(r'/\.(deleted|archived)', '', destpath)
-
- if dest_path == self.path:
- ManageActions(
- action_type=ActionDescription.DELETION if '.deleted' in destpath else ActionDescription.ARCHIVING,
- api=self.content_api,
- content=self.content,
- session=self.session,
- ).action()
- else:
- invalid_path = True
- # else we check if the path is good (not at the root path / not in a deleted/archived path)
- # and we move the content
- else:
- invalid_path = any(x in destpath for x in ['.deleted', '.archived'])
- invalid_path = invalid_path or any(x in self.path for x in ['.deleted', '.archived'])
- invalid_path = invalid_path or dirname(destpath) == self.environ['http_authenticator.realm']
-
- if not invalid_path:
- self.move_file(destpath)
-
- if invalid_path:
- raise DAVError(HTTP_FORBIDDEN)
-
- def move_file(self, destpath: str) -> None:
- """
- Move file mean changing the path to access to a file. This can mean
- simple renaming(1), moving file from a directory to one another(2)
- but also renaming + moving file from a directory to one another at
- the same time (3).
-
- (1): move /dir1/file1 -> /dir1/file2
- (2): move /dir1/file1 -> /dir2/file1
- (3): move /dir1/file1 -> /dir2/file2
- :param destpath: destination path of webdav move
- :return: nothing
- """
-
- workspace = self.content.workspace
- parent = self.content.parent
-
- with new_revision(
- content=self.content,
- tm=transaction.manager,
- session=self.session,
- ):
- # INFO - G.M - 2018-03-09 - First, renaming file if needed
- if basename(destpath) != self.getDisplayName():
- new_given_file_name = transform_to_bdd(basename(destpath))
- new_file_name, new_file_extension = \
- os.path.splitext(new_given_file_name)
-
- self.content_api.update_content(
- self.content,
- new_file_name,
- )
- self.content.file_extension = new_file_extension
- self.content_api.save(self.content)
-
- # INFO - G.M - 2018-03-09 - Moving file if needed
- workspace_api = WorkspaceApi(
- current_user=self.user,
- session=self.session,
- )
- content_api = ContentApi(
- current_user=self.user,
- session=self.session,
- config=self.provider.app_config
- )
-
- destination_workspace = self.provider.get_workspace_from_path(
- destpath,
- workspace_api,
- )
- destination_parent = self.provider.get_parent_from_path(
- destpath,
- content_api,
- destination_workspace,
- )
- if destination_parent != parent or destination_workspace != workspace: # nopep8
- # INFO - G.M - 12-03-2018 - Avoid moving the file "at the same place" # nopep8
- # if the request does not result in a real move.
- self.content_api.move(
- item=self.content,
- new_parent=destination_parent,
- must_stay_in_same_workspace=False,
- new_workspace=destination_workspace
- )
-
- transaction.commit()
-
- def copyMoveSingle(self, destpath, isMove):
- if isMove:
- # INFO - G.M - 12-03-2018 - This case should not happen
- # As far as moveRecursive method exist, all move should not go
- # through this method. If such case appear, try replace this to :
- ####
- # self.move_file(destpath)
- # return
- ####
-
- raise NotImplemented
-
- new_file_name = None
- new_file_extension = None
-
- # Inspect destpath
- if basename(destpath) != self.getDisplayName():
- new_given_file_name = transform_to_bdd(basename(destpath))
- new_file_name, new_file_extension = \
- os.path.splitext(new_given_file_name)
-
- workspace_api = WorkspaceApi(
- current_user=self.user,
- session=self.session,
- )
- content_api = ContentApi(
- current_user=self.user,
- session=self.session,
- config=self.provider.app_config
- )
- destination_workspace = self.provider.get_workspace_from_path(
- destpath,
- workspace_api,
- )
- destination_parent = self.provider.get_parent_from_path(
- destpath,
- content_api,
- destination_workspace,
- )
- workspace = self.content.workspace
- parent = self.content.parent
- new_content = self.content_api.copy(
- item=self.content,
- new_label=new_file_name,
- new_parent=destination_parent,
- )
- self.content_api.copy_children(self.content, new_content)
- transaction.commit()
-
- def supportRecursiveMove(self, destPath):
- return True
-
- def delete(self):
- ManageActions(
- action_type=ActionDescription.DELETION,
- api=self.content_api,
- content=self.content,
- session=self.session,
- ).action()
-
-
- class HistoryFileResource(FileResource):
- """
- A virtual resource corresponding to a specific tracim's revision's file
- """
- def __init__(self, path: str, environ: dict, content: Content, user: User, session: Session, content_revision: ContentRevisionRO):
- super(HistoryFileResource, self).__init__(path, environ, content, user=user, session=session)
- self.content_revision = content_revision
-
- def __repr__(self) -> str:
- return "<DAVNonCollection: HistoryFileResource (%s-%s)" % (self.content.content_id, self.content.file_name)
-
- def getDisplayName(self) -> str:
- left_side = '(%d - %s) ' % (self.content_revision.revision_id, self.content_revision.revision_type)
- return '%s%s' % (left_side, transform_to_display(self.content_revision.file_name))
-
- def getContent(self):
- filestream = compat.BytesIO()
- filestream.write(self.content_revision.depot_file.file.read())
- filestream.seek(0)
-
- return filestream
-
- def getContentLength(self):
- return self.content_revision.depot_file.file.content_length
-
- def getContentType(self) -> str:
- return self.content_revision.file_mimetype
-
- def beginWrite(self, contentType=None):
- raise DAVError(HTTP_FORBIDDEN)
-
- def delete(self):
- raise DAVError(HTTP_FORBIDDEN)
-
- def copyMoveSingle(self, destpath, ismove):
- raise DAVError(HTTP_FORBIDDEN)
-
-
- class OtherFileResource(FileResource):
- """
- FileResource resource corresponding to tracim's page and thread
- """
- def __init__(self, path: str, environ: dict, content: Content, user:User, session: Session):
- super(OtherFileResource, self).__init__(path, environ, content, user=user, session=session)
-
- self.content_revision = self.content.revision
-
- self.content_designed = self.design()
-
- # workaround for consistent request as we have to return a resource with a path ending with .html
- # when entering folder for windows, but only once because when we select it again it would have .html.html
- # which is no good
- if not self.path.endswith('.html'):
- self.path += '.html'
-
- def getDisplayName(self) -> str:
- return self.content.get_label_as_file()
-
- def getPreferredPath(self):
- return self.path
-
- def __repr__(self) -> str:
- return "<DAVNonCollection: OtherFileResource (%s)" % self.content.file_name
-
- def getContentLength(self) -> int:
- return len(self.content_designed)
-
- def getContentType(self) -> str:
- return 'text/html'
-
- def getContent(self):
- filestream = compat.BytesIO()
-
- filestream.write(bytes(self.content_designed, 'utf-8'))
- filestream.seek(0)
- return filestream
-
- def design(self):
- if self.content.type == ContentType.Page:
- return designPage(self.content, self.content_revision)
- else:
- return designThread(
- self.content,
- self.content_revision,
- self.content_api.get_all(self.content.content_id, ContentType.Comment)
- )
-
-
- class HistoryOtherFile(OtherFileResource):
- """
- A virtual resource corresponding to a specific tracim's revision's page and thread
- """
- def __init__(self,
- path: str,
- environ: dict,
- content: Content,
- user:User,
- content_revision: ContentRevisionRO,
- session: Session):
- super(HistoryOtherFile, self).__init__(
- path,
- environ,
- content,
- user=user,
- session=session
- )
- self.content_revision = content_revision
- self.content_designed = self.design()
-
- def __repr__(self) -> str:
- return "<DAVNonCollection: HistoryOtherFile (%s-%s)" % (self.content.file_name, self.content.id)
-
- def getDisplayName(self) -> str:
- left_side = '(%d - %s) ' % (self.content_revision.revision_id, self.content_revision.revision_type)
- return '%s%s' % (left_side, transform_to_display(self.content_revision.get_label_as_file()))
-
- def getContent(self):
- filestream = compat.BytesIO()
-
- filestream.write(bytes(self.content_designed, 'utf-8'))
- filestream.seek(0)
-
- return filestream
-
- def delete(self):
- raise DAVError(HTTP_FORBIDDEN)
-
- def copyMoveSingle(self, destpath, ismove):
- raise DAVError(HTTP_FORBIDDEN)
|