mirror of
https://github.com/klzgrad/naiveproxy.git
synced 2024-12-01 01:36:09 +03:00
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
# Copyright 2017 The Chromium Authors. All rights reserved.
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
|
|
import time
|
|
import os
|
|
import shutil
|
|
import urllib
|
|
|
|
from pylib.base import output_manager
|
|
|
|
|
|
class LocalOutputManager(output_manager.OutputManager):
|
|
"""Saves and manages test output files locally in output directory.
|
|
|
|
Location files will be saved in {output_dir}/TEST_RESULTS_{timestamp}.
|
|
"""
|
|
|
|
def __init__(self, output_dir):
|
|
super(LocalOutputManager, self).__init__()
|
|
timestamp = time.strftime(
|
|
'%Y_%m_%dT%H_%M_%S', time.localtime())
|
|
self._output_root = os.path.abspath(os.path.join(
|
|
output_dir, 'TEST_RESULTS_%s' % timestamp))
|
|
|
|
#override
|
|
def _CreateArchivedFile(self, out_filename, out_subdir, datatype):
|
|
return LocalArchivedFile(
|
|
out_filename, out_subdir, datatype, self._output_root)
|
|
|
|
|
|
class LocalArchivedFile(output_manager.ArchivedFile):
|
|
|
|
def __init__(self, out_filename, out_subdir, datatype, out_root):
|
|
super(LocalArchivedFile, self).__init__(
|
|
out_filename, out_subdir, datatype)
|
|
self._output_path = os.path.join(out_root, out_subdir, out_filename)
|
|
|
|
def _Link(self):
|
|
return 'file://%s' % urllib.quote(self._output_path)
|
|
|
|
def _Archive(self):
|
|
if not os.path.exists(os.path.dirname(self._output_path)):
|
|
os.makedirs(os.path.dirname(self._output_path))
|
|
shutil.copy(self.name, self._output_path)
|