casa6

import os

import fnmatch

import subprocess

import tarfile

from casac import casac

import datetime

import time

import urllib2

import __casac__

import TelemetryLogMonitor

import ssl

​

class telemetry:

​

    def __init__(self, casa):

​

        self.setCasaVersion()

        self.setHostId()

        self.logdir = casa['dirs']['rc']

        self.variantSuffix = ""

        if len(casa['variant'])>1:

            self.variantSuffix = "-" + casa['variant']

        self.logpattern = 'casastats-' + self.casaver + '-' + self.hostid + '*' + self.variantSuffix + '.log'

        self.sendlogpattern = 'casastats-*'+ self.hostid + '*.log'

        self.stampfile = self.logdir + '/telemetry-' + self.hostid + '.stamp'

        self.casa = casa

​

        logfiles = []

​

        # Check if user has defined a telemetry log location

        casa_util = __casac__.utils.utils()

​

        if (casa_util.getrc("TelemetryLogDirectory") != 'Unknown value'):

            self.logdir = casa_util.getrc("TelemetryLogDirectory")

​

        for file in os.listdir(self.logdir):

            if fnmatch.fnmatch(file, self.logpattern):

                 #print "Matched: " + file

                 logfiles.append(file)

​

        logfiles.sort(reverse=True)

        # Size of the existing (non-active) logfiles

        inactiveTLogSize = 0

​

        if (logfiles and logfiles[0] != None):

            print "Found an existing telemetry logfile: " + self.logdir  + "/" + logfiles[0]

            casa['files']['telemetry-logfile'] = self.logdir  + "/" + logfiles[0]

            for i in range(1, len(logfiles)):

                inactiveTLogSize = inactiveTLogSize + os.path.getsize(self.logdir  + "/" + logfiles[i])/1024

                #print "Inactive log size: " + str(inactiveTLogSize)

        else :

            print "Creating a new telemetry file"

            self.setNewTelemetryFile()

​

        # Setup Telemetry log size monitoring

        # Size limit for the telemetry logs

        tLogSizeLimit = 20000

        # File size check interval

        tLogSizeInterval = 60

        try:

            tLogSizeLimit = int(casa_util.getrc("TelemetryLogLimit"))

            tLogSizeInterval = int(casa_util.getrc("TelemetryLogSizeInterval"))

        except:

            pass

        # Subtract the inactive log sizes from the total log file size limit

        tLogSizeLimit = tLogSizeLimit - inactiveTLogSize

        if (tLogSizeLimit <= 0):

            print "Telemetry log size limit exceeded. Disabling telemetry."

            casa['state']['telemetry-enabled'] = False

        else :

            tLogMonitor = TelemetryLogMonitor.TelemetryLogMonitor()

            tLogMonitor.start(casa['files']['telemetry-logfile'],tLogSizeLimit, tLogSizeInterval, casa)

            print "Telemetry initialized. Telemetry will send anonymized usage statistics to NRAO."

            print 'You can disable telemetry by adding the following line to your ~/.casarc file:'

            print 'EnableTelemetry: False'

​

    def setNewTelemetryFile(self):

        self.casa['files']['telemetry-logfile'] =  self.logdir + '/casastats-' + self.casaver +'-'  + self.hostid + "-" + time.strftime("%Y%m%d-%H%M%S", time.gmtime()) + self.variantSuffix + '.log'

        # Work around the chicken/egg problem with telemetry/logger initialization

        if hasattr(self, 'logger'):

            self.logger.setstatslogfile(self.casa['files']['telemetry-logfile'])

​

    def setCasaVersion(self):

        myUtils = casac.utils()

        ver = myUtils.version()

        self.casaver = str(ver[0])+ str(ver[1]) + str(ver[2])+ "-" + str(ver[3])

​

    def setHostId(self):

        telemetryhelper = casac.telemetryhelper()

        self.hostid = telemetryhelper.getUniqueId()

​