def get_rtree(date,
hdfs_rtree_path="/user/hadoop/rtree/shanghai/%s-%s/%s.%s",
local_path="/home/hadoop/data/downloaded/rtree/%s-%s/"):
year, month, day = date.split('-')
# guarantee there exists local directory for the worker nodes
path = local_path % (year, month)
if not os.path.exists(path):
os.makedirs(path)
client = Client(QuerierParallel.master_hdfs_path,
root="/",
timeout=100,
session=False)
client.download(hdfs_rtree_path % (year, month, day, "data"),
path,
overwrite=True)
client.download(hdfs_rtree_path % (year, month, day, "index"),
path,
overwrite=True)
# contruct the rtree
rtree_properties = index.Property()
rtree_properties.dat_extension = 'data'
rtree_properties.idx_extension = 'index'
rtree = index.Index(path + day, properties=rtree_properties)
return rtree
class HdfsClient(object):
def __init__(self, url=None):
self.url = url
self.client = Client(url=url)
def ls(self, path):
return self.client.list(path)
def isFile(self, path):
result = self.client.status(path, strict=False)
if result:
return result[TYPE] == FILE
else:
return False
def mkdir(self, path):
self.client.makedirs(path, permission=777)
def isDirectory(self, path):
result = self.client.status(path, strict=False)
if result:
return result[TYPE] == DIRECTORY
else:
return False
def upload(self, localSourcePath, remoteDistPath):
self.client.upload(remoteDistPath, localSourcePath, overwrite=True)
def dowload(self, remoteSourcePath, localDistPath):
self.client.download(remoteSourcePath, localDistPath, overwrite=True)
def put(self, localSourcePath, remoteDistPath):
with open(localSourcePath,
"r") as reader, self.client.write(remoteDistPath) as writer:
data = reader.read(FILE_SIZE)
while data != "":
writer.write(data)
data = reader.read(FILE_SIZE)
def get(self, remoteSourcePath, localDistPath):
with self.client.read(remoteSourcePath,
chunk_size=FILE_SIZE) as reader, open(
localDistPath, "a+") as writer:
for chunk in reader:
writer.write(chunk)
def get(self, request):
"""
计算结果下载hdfs 文件
:param request:
:return:
"""
hdfsPath = request.GET.get("hdfsPath")
logger.debug("请求文件：{0}".format(hdfsPath))
localPath = os.path.join(settings.BASE_DIR, 'media', 'hdfsFile')
logger.debug("本地存储路径：{0}".format(localPath))
# 链接HDFS下载文件
cli = Client(settings.HDFS_HOST)
logger.debug("HDFS连接{0}".format(cli))
try:
fileName = cli.list(hdfsPath)[1]
# print("filename:", fileName)
path = os.path.join(hdfsPath, fileName)
logger.debug(path, localPath)
cli.download(hdfs_path=path, local_path=localPath, overwrite=True)
except HdfsError:
return Response(data={"error": "文件未找到"},
status=status.HTTP_404_NOT_FOUND)
return Response(data={"fileName": fileName}, status=status.HTTP_200_OK)
print("hdfs中的目录为:", client.list(hdfs_path="/test1", status=True))
#read()
#读取文件信息 类似与 hdfs dfs -cat hfds_path,参数如下:
# hdfs_path hdfs路径
# offset 读取位置
# length 读取长度
# buffer_size 设置buffer_size 不设置使用hdfs默认100MB 对于大文件 buffer够大的化 sort与shuffle都更快
# encoding 指定编码
# chunk_size 字节的生成器,必须和encodeing一起使用 满足chunk_size设置即 yield
# delimiter 设置分隔符 必须和encodeing一起设置
# progress 读取进度回调函数 读取一个chunk_size回调一次
print(
client.read(
"/test1/part-00000-15b6e708-1025-408b-a6f2-1f37a7fe7064-c000.csv"))
# 下载
print(
"下载文件结果part.csv:",
client.download(
hdfs_path=
"/test1/part-00000-15b6e708-1025-408b-a6f2-1f37a7fe7064-c000.csv",
local_path="/home/yyj2020/test",
overwrite=True))
# 打包
with tarfile.open("/home/yyj2020/test/tartest.tar.gz", "w:gz") as tar:
tar.add("/home/yyj2020/test/files",
arcname=os.path.basename("/home/yyj2020/test/files"))
tar.close()
' a study.__init__.py module '
__author__ = 'steven'
import os
import time
from hdfs.client import Client
client = Client("http://127.0.0.1:50070", root="/", timeout=100)
print(client.makedirs("/test/"))
print(client.status("/test/"))
print(client.list("/test/"))
print(client.delete("/test/", True))
upload_filename = client.upload(
"/test/" + str(int(round(time.time() * 1000))) + ".pdf", "test.pdf")
print(upload_filename)
download_path = os.path.join(os.path.abspath('.'), 'download/hdfs/')
if not os.path.exists(download_path):
os.makedirs(download_path, True)
else:
print(download_path, ' is existed.')
print(
client.download(
upload_filename,
download_path + str(int(round(time.time() * 1000))) + ".pdf"))
print(client.delete(upload_filename))
print(client.delete(upload_filename))
import sys
import time
from importlib import reload
from hdfs.client import Client
import urllib.request
reload(sys)
url = 'http://localhost:50070'
client = Client(url=url)
path = "/pyhdfs/"
# p = client.resolve(path)
# print(p)
# client.makedirs(hdfs_path=path)
# l_path= "2020-03-10-23-01-4.txt"
# client.upload(hdfs_path=path, local_path=l_path)
client.download(path + "2020-03-10-23-01-4.txt", '.')
d = time.strftime("%Y-%m", time.localtime())
print(d)
urllib.request.urlretrieve()
def do2():
global csv_path
client = Client(hdfshost)
client.download(csv_path, local_file_path)
class HDFSClient:
def __init__(self,
url,
root=None,
user=None,
proxy=None,
timeout=None,
session=None):
""" 连接hdfs
url: HDFS名称节点的主机名或IP地址及端口号
root: 根路径，此路径将作为传递给客户端的所有HDFS路径的前缀
user: 使用InsecureClient（Base Client）,指定访问hdfs的用户;Client使用默认用户dr.who
proxy: 代理的用户
timeout: 连接超时，转发到请求处理程序
session: request.Session实例，用于发出所有请求
"""
if user:
self.client = InsecureClient(url, user=user)
else:
self.client = Client(url,
root=root,
proxy=proxy,
timeout=timeout,
session=session)
def list_hdfs_file(self, hdfs_path, status=False):
""" 返回目录下的文件
status: 每个文件或目录的属性信息(FileStatus)
return: 列表中包含元组，每个元组是目录名或文件名和属性信息构成
"""
return self.client.list(hdfs_path, status=status)
def walk_hdfs_file(self,
hdfs_path,
depth=0,
status=False,
ignore_missing=False,
allow_dir_changes=False):
""" 深度遍历远程文件系统
hdfs_path: 起始路径。如果该路径不存在，则会引发HdfsError。如果指向文件，则返回的生成器将为空
depth: 探索的最大深度。0为无限制
status: 同时返回每个文件或文件夹的相应FileStatus
ignore_missing: 忽略缺少的嵌套文件夹，而不是引发异常
allow_dir_changes: 允许更改目录列表以影响遍历
return: 生成器，返回值参考python的walk函数
"""
return self.client.walk(hdfs_path,
depth=depth,
status=status,
ignore_missing=ignore_missing,
allow_dir_changes=allow_dir_changes)
def delete_hdfs_file(self, hdfs_path, recursive=False, skip_trash=False):
""" 删除文件
recursive: 递归删除文件或目录，默认情况下，如果尝试删除非空目录，此方法将引发HdfsError
skip_trash: 设置为false时，已删除的路径将被移动到适当的垃圾回收文件夹，而不是被删除
return: 如果删除成功，则此函数返回True；如果hdfs_path之前不存在文件或目录，则返回False
"""
return self.client.delete(hdfs_path,
recursive=recursive,
skip_trash=skip_trash)
def download_hdfs_file(self,
hdfs_path,
local_path,
overwrite=True,
n_threads=1,
temp_dir=None,
**kwargs):
""" 下载文件
hdfs_file: HDFS上要下载的文件或文件夹的路径。如果是文件夹，则将下载该文件夹下的所有文件
local_file: 本地路径。如果它已经存在并且是目录，则文件将在其中下载
overwrite: 覆盖任何现有文件或目录
n_threads: 用于并行化的线程数。值为0（或负数）将使用与文件一样多的线程
temp_dir: 当overwrite = True并且最终目标路径已经存在时，将首先在其下下载文件的目录。下载成功完成后，它将被交换
**kwargs: 关键字参数转发给read()。如果未传递chunk_size参数，则将使用默认值64 kB
return: 方法执行成功，将返回本地下载路径
"""
res = self.client.download(hdfs_path,
local_path,
overwrite=overwrite,
n_threads=n_threads,
temp_dir=temp_dir,
**kwargs)
def upload_hdfs_file(self,
hdfs_path,
local_path,
n_threads=1,
temp_dir=None,
chunk_size=65536,
progress=None,
cleanup=True,
**kwargs):
""" 上传文件
hdfs_path: 目标HDFS路径。如果它已经存在并且是目录，则文件将在其中上传
local_path: 文件或文件夹的本地路径。如果是文件夹，则将上载其中的所有
文件（请注意，这意味着没有文件的文件夹将不会远程创建）
cleanup: 如果上传过程中发生错误，删除所有上传的文件
return: 方法执行成功，将返回状态码，远程上传目录，错误信息
"""
try:
res = self.client.upload(hdfs_path,
local_path,
n_threads=n_threads,
temp_dir=temp_dir,
chunk_size=chunk_size,
progress=progress,
cleanup=cleanup,
overwrite=True)
return 0, res, ''
except HdfsError as e:
return 1, '', str(e)
def makedirs(self, hdfs_path, permission=None):
""" 创建目录，可以递归
permission: 在新创建的目录上设置的八进制权限，这些权限将仅在尚不存在的目录上设置
return: None
"""
self.client.makedirs(hdfs_path, permission=permission)
def parts(self, hdfs_path, parts=None, status=False):
"""
hdfs_path: 远程路径。该目录每个分区最多应包含一个零件文件（否则将任意选择一个文件）
parts: 零件文件编号列表或要选择的零件文件总数。如果是数字，那么将随机选择那么多分区。
默认情况下，将返回所有零件文件。如果部件是列表，但未找到部件之一或需要太多样本，则会引发HdfsError
status: 返回文件的FileStatus
return: 返回对应于路径的零件文件的字典
"""
return self.client.parts(hdfs_path, parts=parts, status=status)
def read_hdfs_file(self, **kwds):
""" 读取文件内容，这个方法必须在一个with块中使用，以便关闭连接
>>> with client.read('foo') as reader:
>>> content = reader.read()
hdfs_path: HDFS路径
offset: 起始字节位置
length: 要处理的字节数。设置为None时会读取整个文件
buffer_size: 用于传输数据的缓冲区大小（以字节为单位）。默认为在HDFS配置中设置的值
encoding: 用于解码请求的编码。默认情况下，返回原始数据
chunk_size: 如果设置为正数，则上下文管理器将返回一个生成器，该生成器生成每个chunk_size字节，
而不是类似文件的对象（除非还设置了定界符）
delimiter: 如果设置，上下文管理器将在每次遇到定界符时返回生成器。此参数要求指定编码
progress: 回调函数，用于跟踪进度，称为每个chunk_size字节（如果未指定块大小，则不可用）。
它将传递两个参数，即要上传的文件的路径和到目前为止已传输的字节数。
完成后，将以-1作为第二个参数调用一次
"""
self.client.read(**kwds)
def write_hdfs_file(self,
hdfs_path,
data=None,
overwrite=False,
permission=None,
blocksize=None,
replication=None,
buffersize=None,
append=False,
encoding=None):
""" 在HDFS上创建文件
data: 要写入的文件内容。 可以是字符串，生成器或文件对象。 最后两个选项将允许流式上传（即无需
将全部内容加载到内存中）。 如果为None，则此方法将返回类似文件的对象，应使用with块调用
它（请参见下面的示例）
permission: 在新创建的文件上设置的八进制权限
append: 附加到文件而不是创建新文件
encoding: 用于序列化写入数据的编码
>>> from json import dump, dumps
>>> records = [
>>> {'name': 'foo', 'weight': 1},
>>> {'name': 'bar', 'weight': 2},
>>> ]
>>> # As a context manager:
>>> with client.write('data/records.jsonl', encoding='utf-8') as writer:
>>> dump(records, writer)
>>> Or, passing in a generator directly:
>>> client.write('data/records.jsonl', data=dumps(records), encoding='utf-8')
"""
self.client.write(hdfs_path,
data=data,
overwrite=overwrite,
permission=permission,
blocksize=blocksize,
replication=replication,
buffersize=buffersize,
append=append,
encoding=encoding)
def rename_or_move(self, hdfs_src_path, hdfs_dst_path):
""" 移动文件或目录
hdfs_src_path: 源路径
hdfs_dst_path: 目标路径，如果路径已经存在并且是目录，则源将移入其中。
如果路径存在并且是文件，或者缺少父目标目录，则此方法将引发HdfsError
"""
self.client.rename(hdfs_src_path, hdfs_dst_path)
def set_owner(self, hdfs_path, owner=None, group=None):
""" 更改文件的所有者，必须至少指定所有者和组之一
owner: 可选，文件的新所有者
group: 可选，文件的新所有组
"""
self.client.set_owner(hdfs_path, owner=owner, group=group)
def set_permission(self, hdfs_path, permission):
""" 更改文件权限
permission: 文件的新八进制权限字符串
"""
self.client.set_permission(hdfs_path, permission)
def set_replication(self, hdfs_path, replication):
""" 设置文件副本
replication: 副本数
"""
self.client.set_replication(hdfs_path, replication)
def set_times(self, hdfs_path, access_time=None, modification_time=None):
""" 更改文件的时间戳
"""
self.client.set_times(hdfs_path,
access_time=access_time,
modification_time=modification_time)
def status_hdfs_file(self, hdfs_path, strict=True):
""" 获取文件的FileStatus
strict: 如果为False，则返回None，而不是如果路径不存在则引发异常
"""
self.client.status(hdfs_path, strict=strict)
#!/usr/bin/evn python
# -*- coding: utf-8 -*-
# python version 2.7.6
'''
此例子使用的是hdfs模块
'''
from hdfs.client import Client
myClient = Client("http://study:50070")
myClient.content("/data/gz")
hdfsPath = "/data/gz/hadoop-2.7.1.tar.gz"
localPath = "/root/data/"
#从hdfs中取出数据存放到本地系统硬盘上
myClient.download(hdfsPath,localPath)
'''
此部分代码没有走通
fq = "/data/fq/20150923.fasta"
f = myClient.read(fq)
for line in f:
print line
'''