import json
from datetime import datetime as dt

from helpers.utils import get_logger, get_raw_json, get_interim_file

logger = get_logger(__name__)


def build_search_info(data):
    return {
        'buildVersion':
        data.get('applicationProperties').get('info.build.version'),
        'isBot':
        data.get('bot'),
        'deviceType':
        data.get('deviceType'),
        'listViewUrl':
        data.get('listViewUrl'),
        'resultPages':
        data.get('pagination').get('total'),
        'searchDate':
        parse_timestamp(data.get('recentSearchModel').get('createDate')),
        'resultCount':
        int(data.get('resultCount')),
        'timestamp':
        parse_timestamp(data.get('timestamp'))
    }


def build_list_result_info(item, search_info):
    return {
示例#2
0
import tempfile
import requests
from pydrive.drive import GoogleDrive
from helpers.utils import IVLE_DOWNLOAD_URL, get_logger

logger = get_logger()


def create_folder(drive, name, parent_id=None):
    folder = drive.CreateFile({
        'title': name,
        'mimeType': 'application/vnd.google-apps.folder',
    })
    if parent_id:
        folder['parents'] = [{'kind': 'drive#fileLink', "id": parent_id}]
    folder.Upload()
    return folder


def create_file(drive, name, content, parent_id=None):
    file = drive.CreateFile({"title": name})
    tf = tempfile.NamedTemporaryFile('rb+')
    tf.write(content)
    tf.seek(0)
    file.SetContentFile(tf.name)
    if parent_id:
        file['parents'] = [{'kind': 'drive#fileLink', "id": parent_id}]
    file.Upload()
    return file

import pandas as pd
import numpy as np
from datetime import date as dt

from helpers import utils
from app import company_files as company

logger = utils.get_logger(__name__)

colOrder = [
    'loaddate', 'listdate', 'market', 'company', 'sector', 'subsector',
    'marketcap', 'country'
]


def load_excel_sheet(load_date, load_definition, sheet_idx, sheet_definition):
    return (pd.read_excel(utils.get_project_path('data', 'external',
                                                 'company-files',
                                                 load_definition['filename']),
                          sheet_name=sheet_idx,
                          header=None,
                          names=sheet_definition['colNames'],
                          usecols=sheet_definition['parseCols'],
                          skiprows=sheet_definition['headerRow']).assign(
                              loaddate=pd.to_datetime(load_date))[colOrder])


def load_all():
    data = []
    missed_sheets = []
示例#4
0
import helpers.vmr.a8.a8_helper as a8
import helpers.models.recording as recording_model
import helpers.notification.utils as notification_utils
import validators.validate_common as validate_common
import validators.validate_recordings as validate_recordings
import validators.validate_storage as validate_storage

from helpers.constants import Cos, Component
from helpers.constants import Stream
from helpers.constants import TestLog
from helpers.constants import TimeFormat
from helpers.constants import ValidationError
from helpers.constants import RecordingAttribute

pytestmark = pytest.mark.recording
LOGGER = utils.get_logger()


def test_deleted_recording_cannot_be_played_back_ipdvrtests_166(
        common_lib, stream):
    """
    JIRA ID : IPDVRTESTS-166
    JIRA LINK : https://jira01.engit.synamedia.com/browse/IPDVRTESTS-166
    """
    recording = None
    web_service_obj = None

    try:
        # Step1: Create a 30 minute recording
        recording, web_service_obj = common_lib.create_recording(
            stream, rec_duration=Component.LARGE_REC_LEN_IN_SEC)
示例#5
0
from kubernetes import config
from test_manifests import test_manifest

import helpers.setup as test_setup
import helpers.teardown as test_teardown
import helpers.utils as utils
import helpers.constants as constants

from helpers import dp_lib
from helpers.constants import Component, VMR_KUBECONFIG
from helpers.constants import PytestOptions
from helpers.constants import RecordingAttribute
from helpers.constants import TestLog, Stream_tags
from helpers.constants import TestDirectory, TestReport

LOGGER = utils.get_logger()
debug_logger = utils.get_logger(TestLog.DEBUG_LOGGER)
# log('test_manifest.tests=%s' % test_manifest.tests)
# debug_logger.info('test_manifest.tests=%s' % test_manifest.tests)

prefix_on_log_errors = 'XX' * 25
suffix_on_log_errors = 'XX' * 25

orders_map = {
    'first': 0,
    'second': 1,
    'third': 2,
    'fourth': 3,
    'fifth': 4,
    'sixth': 5,
    'seventh': 6,