def shorten(self, user_input): """ Identifies the type of the input Whether it's a file or a url and calls the method to shorten the url @:param: the input that user provided """ input_type = findType.find_argument_type(user_input) if input_type == 'file': logger.info('Given input is a file') content = util.read_file(user_input) index = 1 for url in content: if index % 10000 == 0: logger.info('finished : {value}'.format(value=index)) short_url = self.shorten_url(url) logger.info('Short url : {url}'.format(url=short_url)) original_url = self.redirect(short_url) logger.info('Re-directing to : {url}'.format(url=original_url)) index += 1 else: logger.info('Given input is a URL') short_url = self.shorten_url(user_input) logger.info('Short url : {url}'.format(url=short_url)) original_url = self.redirect(short_url) logger.info('Re-directing to : {url}'.format(url=original_url))
def test_url_shortening_with_file_as_input(self): url_shorten = UrlShorten() test_file_path = 'test_links.txt' content = my_util.read_file(test_file_path) for url in content: short_url = url_shorten.shorten_url(url) original_url = url_shorten.redirect(short_url) self.assertTrue(url, original_url)
from tools.redmine import Redmine from tools.svn_edge import SvnEdge from tools.mssql import Mssql from user import User from utils.config_utils import ConfigUtils from utils.mail_utils import Mailer from utils.template_utils import Template if __name__ == '__main__': parser = argparse.ArgumentParser(description="Echo account -i id -n name") parser.add_argument('-i', help="develper_id", required=True) parser.add_argument('-n', help="develper_name", required=True) args = parser.parse_args() config = ConfigUtils().get_config() email_config = config['email'] eamil_addr = args.i + '@' + email_config['service-host'] #user 객체 새성 user = User(args.i, args.n, eamil_addr) #계정 생성이 필요한 서비스를 list에 넣는다. services = [] services.append(SvnEdge()) services.append(Redmine()) services.append(Mssql()) #성공과 실패를 분리 success = []
import os import sys # 부모디렉토리 참조를 위한 설정추가 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) from tools.redmine import Redmine from utils.config_utils import ConfigUtils from user import User # given config = ConfigUtils().set_config_path('../config.yml').get_config() user = User('test001', '홍길동', '*****@*****.**') redmine = Redmine() redmine.set_config(config) # when 계정생성 redmine._create_user(user) # when - get sync new_id = redmine._find_user(user.get_id()) print(new_id) # when - create account redmine._add_project_user(new_id) # then - exception이 발생하지 않았으면 성공 print('테스트 성공')
import argparse import logging as logger import os import uuid from collections import OrderedDict from logging.handlers import WatchedFileHandler from urllib.parse import urlparse # Python 3 import utils.sql_query_builder as builder import utils.sql_query_executor as executor from utils.config_utils import ConfigUtils as util from utils.find_argument_type import FindArgumentType as findType str_encode = str.encode configs = util.read_file('/resources/properties.yaml') os.makedirs(configs['log_file_directory'], exist_ok=True) LOG_FILENAME = configs['log_file_directory'] + 'logs.txt' """ Setting the date format for log message """ date_str = "%m/%d/%Y %I:%M:%S %p" """ Adding Handler Filter for Adding Host Name """ stream_handler = logger.StreamHandler() file_handler = WatchedFileHandler(LOG_FILENAME) """ Setting up the format of the log message. """
# coding:utf-8 import os import re from collections import Counter from copy import deepcopy from os import walk import xlsxwriter # from ftpdownload3 import FtpDownload import ftpdownload from utils.config_utils import ConfigUtils from utils.time_utils import TimeUtils # 读取配置文件 caseMaps, appMaps, deviceMaps = ConfigUtils.readConfig() # print caseMaps, appMaps, deviceMaps # print "caseMaps", caseMaps today = TimeUtils.getTodayTimeStamp() today = "170610" # 170610115939 def parse_logs(): # ftpdown = ftpdownload3.FtpDownload() # ftpdown.download("172.25.105.226", "UniversalTest/performance_ScenarioTest/", "20170610115900") # 6.7.1//7.0.0 dir1 = "logs/PD1619/" + today + "/" dir2 = "logs/PD1610/" + today + "/"
def __init__(self): config = ConfigUtils().get_config() self._mail_config = mail_config = config[self.__KEY] self._smtp_host = mail_config['smtp-host'] self._login_id = mail_config['login-id'] self._login_pw = mail_config['login-pw']