ai-content-maker/.venv/Lib/site-packages/smart_open/local_file.py

44 lines
1004 B
Python

# -*- coding: utf-8 -*-
#
# Copyright (C) 2020 Radim Rehurek <me@radimrehurek.com>
#
# This code is distributed under the terms and conditions
# from the MIT License (MIT).
#
"""Implements the transport for the file:// schema."""
import io
import os.path
SCHEME = 'file'
URI_EXAMPLES = (
'./local/path/file',
'~/local/path/file',
'local/path/file',
'./local/path/file.gz',
'file:///home/user/file',
'file:///home/user/file.bz2',
)
open = io.open
def parse_uri(uri_as_string):
local_path = extract_local_path(uri_as_string)
return dict(scheme=SCHEME, uri_path=local_path)
def open_uri(uri_as_string, mode, transport_params):
parsed_uri = parse_uri(uri_as_string)
fobj = io.open(parsed_uri['uri_path'], mode)
return fobj
def extract_local_path(uri_as_string):
if uri_as_string.startswith('file://'):
local_path = uri_as_string.replace('file://', '', 1)
else:
local_path = uri_as_string
return os.path.expanduser(local_path)