41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
from typing import Self
|
||
|
|
|
||
|
|
from .FileContext import FileContext
|
||
|
|
|
||
|
|
class CopyContext:
|
||
|
|
|
||
|
|
def __init__(self, src_uri: str, dst_uri: str, chroot=False) -> None:
|
||
|
|
self.__src_uri = src_uri
|
||
|
|
self.__dst_uri = dst_uri
|
||
|
|
self.__chroot = chroot
|
||
|
|
|
||
|
|
async def __aenter__(self) -> Self:
|
||
|
|
self.__src = FileContext.create(self.__src_uri, chroot=self.__chroot)
|
||
|
|
await self.__src.open()
|
||
|
|
self.__dst = FileContext.create(self.__dst_uri, chroot=self.__chroot)
|
||
|
|
await self.__dst.open()
|
||
|
|
return self
|
||
|
|
|
||
|
|
async def __aexit__(self, exc_type, exc, tb):
|
||
|
|
if self.__src is not None:
|
||
|
|
await self.__src.close()
|
||
|
|
self.__src = None
|
||
|
|
if self.__dst is not None:
|
||
|
|
await self.__dst.close()
|
||
|
|
self.__dst = None
|
||
|
|
|
||
|
|
@property
|
||
|
|
def src(self) -> FileContext:
|
||
|
|
return self.__src
|
||
|
|
|
||
|
|
@property
|
||
|
|
def dst(self) -> FileContext:
|
||
|
|
return self.__dst
|
||
|
|
|
||
|
|
async def _run(self) -> None:
|
||
|
|
await self._run()
|
||
|
|
|
||
|
|
async def run(self) -> None:
|
||
|
|
await self._run()
|