def clearFiles(path, additionalExceptList=[]):
logging.info('Clearing the files in '+str(path)+' ...')
filesInPath = NxFiles.listDir(path)
for i in filesInPath:
if i not in exceptList and i not in additionalExceptList:
NxFiles.removeForce(path+'/'+str(i))
logging.warning(path+'/'+str(i)+' has been removed!')
def _getAndWriteToken(self):
self.logger.debug('Starting to get access_token now...')
self.sender.resetSession()
isQueryOk = False
for i in xrange(self.retry):
try:
accessTokenResponse = self.sender.get(
'https://qyapi.weixin.qq.com/cgi-bin/gettoken?corpid=' + str(self.corpId) + '&corpsecret=' + str(self.secret))
isQueryOk = True
break
except:
self.logger.warning('Getting access_token meets exception!')
e = sys.exc_info()[0]
self.logger.debug(str(e))
isQueryOk = False
time.sleep(1)
if not isQueryOk:
return False
self.logger.debug('The response code is:')
self.logger.debug(str(accessTokenResponse))
myAccessTokenDict = accessTokenResponse.json()
self.logger.debug('The response json is:')
self.logger.debug(str(myAccessTokenDict))
if 'access_token' not in myAccessTokenDict:
self.logger.info('Getting access_token fails!')
if 'errcode' in myAccessTokenDict:
self.logger.info('errcode is ' + str(myAccessTokenDict['errcode']) + '.')
if 'errmsg' in myAccessTokenDict:
self.logger.info('errmsg is ' + str(myAccessTokenDict['errmsg']) + '.')
return False
if 'expires_in' not in myAccessTokenDict:
self.logger.info('expires_in is not in the response json!')
return False
self.accessTokenDict = myAccessTokenDict
self.accessTokenDict['expiresTime'] = self.getExpireTime(self.accessTokenDict['expires_in'])
if not NxFiles.isDir(self.accessTokenStoragePath):
NxFiles.makeDirs(self.accessTokenStoragePath)
myFile = open(self.accessTokenFile, 'w')
myFile.write(str(self.accessTokenDict['expiresTime']) + '\n')
myFile.write(str(self.accessTokenDict['access_token']) + '\n')
myFile.write(str(self.accessTokenDict['expires_in']) + '\n')
myFile.close()
return True
def checkDirectorySize(path, directorySize):
directorySize = long(directorySize)
logger = logging.getLogger()
if not NxFiles.isDir(path):
return
logger.debug('The size of directory ' + str(path) + ' is ' +
str(NxFiles.dirSize(path)))
if NxFiles.dirSize(path) > directorySize:
logger.info('The size of ' + str(path) + ' is greater than ' +
str(directorySize) +
', apttempting to remove files from early to late.')
filesList = NxFiles.sortDir(path)
count = 0
for i in filesList:
NxFiles.removeForce(path + '/' + i)
logger.warning('The file ' + str(path + '/' + i) +
' has been removed!')
if NxFiles.dirSize(path) <= directorySize:
break
if count >= 100:
break
count += 1
else:
logger.debug('The size of ' + str(path) + ' is lower than ' +
str(directorySize))
def getDepartmentList(self):
# https://qyapi.weixin.qq.com/cgi-bin/agent/list?access_token=ACCESS_TOKEN
self.getAccessToken()
for i in xrange(self.retry):
self.logger.info('Starting to get the departments list ...')
try:
response = self.sender.get('https://qyapi.weixin.qq.com/cgi-bin/agent/list?access_token=' + str(
self.accessTokenDict['access_token']))
if response.status_code == 200:
responseJson = response.json()
if 'errcode' in responseJson:
if int(responseJson['errcode']) == 40014:
self.logger.debug(
'The errcode from Wechat server is 40014, the errmsg is "invalid access_token", starting to get access_token again.')
NxFiles.removeForce(self.accessTokenFile)
self.accessTokenDict = None
self.getAccessToken()
elif int(responseJson['errcode'] == 0):
self.logger.info('Getting the departments list succeeds, the list is:')
self.logger.info(str(responseJson['agentlist']))
return responseJson['agentlist']
else:
self.logger.info(
'Response errcode is "' + str(responseJson['errcode']) + '", the reason is "' + str(
responseJson['errmsg']) + '" .')
return False
else:
self.logger.info('Json of response has no errcode, try it again.')
else:
self.logger.info('The status code of response from Wechat server is ' + str(
response.status_code) + ', rather than 200!')
self.logger.debug('Response content is:')
self.logger.debug(str(response.content))
self.logger.debug('Response json is:')
self.logger.debug(str(response.json()))
return False
except:
e = sys.exc_info()[0]
self.logger.warning('Getting departments list meets exception!')
self.logger.debug(str(e))
def getAccessToken(self):
self.logger.debug('Starting to get access_token ...')
if self.accessTokenDict:
if self.isValid():
self.logger.debug('The access_token in the object is valid.')
return True
if NxFiles.isFile(self.accessTokenFile):
self.logger.debug('Starting to verify the access_token file "' + str(self.accessTokenFile) + '" ...')
isFileValid = False
myTokenFile = open(self.accessTokenFile)
fileExpiresTime = myTokenFile.readline()
if fileExpiresTime:
fileAccessToken = myTokenFile.readline()
if fileAccessToken:
fileExpiresIn = myTokenFile.readline()
if fileExpiresIn:
isFileValid = True
else:
isFileValid = False
else:
isFileValid = False
else:
isFileValid = False
if not isFileValid:
self.logger.debug('Access_token file "' + str(self.accessTokenFile) + '" is invalid.')
return self._getAndWriteToken()
else:
fileExpiresIn = fileExpiresIn.replace('\n', '')
fileAccessToken = fileAccessToken.replace('\n', '')
fileExpiresTime = fileExpiresTime.replace('\n', '')
if int(fileExpiresTime) > int(time.time()):
self.logger.debug('The access_token in access_token file "' + str(self.accessTokenFile) + '" is valid.')
if not self.accessTokenDict:
self.accessTokenDict = {}
self.accessTokenDict['expiresTime'] = int(fileExpiresTime)
self.accessTokenDict['access_token'] = str(fileAccessToken)
self.accessTokenDict['expires_in'] = int(fileExpiresIn)
return True
else:
return self._getAndWriteToken()
else:
self.logger.debug('There is no file "' + str(self.accessTokenFile) + '" .')
return self._getAndWriteToken()
NxFiles.removeForce(path+'/'+str(i))
logging.warning(path+'/'+str(i)+' has been removed!')
logging.info('Clean up the ".pyc" file ...')
RemovePyc.delPycFile()
clearFiles(NxSanfranRunPath)
clearFiles(NxLogPrjctsPath)
clearFiles(NxVarLogPath)
clearFiles(NxSanfranTmpPath)
clearFiles(NxTmpPrjctsPath)
logging.info('Clean up the geo_ip_sub folder ...')
geoIpSubParentFolder = NxUsrLibPath+'/NxCallSystem/ADC/IpLibrary'
if NxFiles.isDir(geoIpSubParentFolder):
if NxFiles.isDir(geoIpSubParentFolder+'/geo_ip_sub'):
NxFiles.removeForce(geoIpSubParentFolder+'/geo_ip_sub')
logging.warning(geoIpSubParentFolder+'/geo_ip_sub'+' has been removed.')
def httpDownload(url,
cookieString,
location,
isEntireConfig=True,
fileName=None,
timeout=60,
logger=None):
if logger is None:
logger = logging.getLogger()
if fileName is None:
if isEntireConfig:
fileName = 'AdcConfig' + time.strftime('%Y%m%d%H%M%S',
time.localtime()) + '.tar'
else:
fileName = 'AdcConfig' + time.strftime('%Y%m%d%H%M%S',
time.localtime())
logger.debug('Checking the location ' + str(location))
if not NxFiles.isDir(location):
logger.debug('The location ' + str(location) +
' is not a directory, attempting to make it.')
NxFiles.makeDirs(location)
logger.debug('The location ' + str(location) + ' has been made.')
else:
logger.debug('The location ' + str(location) +
' is a valid directory.')
if NxFiles.isFile(location + '/' + fileName):
logger.warning('The file ' + location + '/' + fileName +
' exists, delete it now!')
NxFiles.removeForce(location + '/' + fileName)
try:
r = NxRequests()
r.addCookie('adc_session', cookieString)
r.setTimeout(timeout)
r.addHeader(
'Accept',
'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8'
)
r.addHeader('Accept-Encoding', 'gzip, deflate, sdch')
r.addHeader('Accept-Language', 'en-US,en;q=0.8')
r.addHeader('Connection', 'keep-alive')
r.addHeader('Upgrade-Insecure-Requests', '1')
r.addHeader(
'User-Agent',
'Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/53.0.2785.116 Safari/537.36'
)
#r.addHeader('referer', refererString)
logger.info('Starting to download...')
response = r.get(url, stream=True)
if response.status_code != 200:
logger.warning(
'The status code from ADC is not 200, return False!')
#wechat2Me('[AdcPatroller] The status code from ADC is not 200! URL is '+str(url)+' .')
return False
if not response.content:
logger.warning(
'The response from ADC has no content, return False!')
#wechat2Me('[AdcPatroller] The response from ADC has no content! The URL is '+str(url)+' .')
return False
else:
if str(response.content).find(invalidLicenseString) != -1:
logger.warning('The response content contains ' +
invalidLicenseString + ' , return False!')
return False
with open(location + '/' + fileName, 'wb') as f:
for chunk in response.iter_content(chunk_size=1024):
if chunk:
f.write(chunk)
logger.info('The config file has been downloaded to ' + location +
'/' + fileName + ' .')
except Exception as err:
logger.debug('Downloading meete exception:')
logger.debug(str(err))
return False
def forceMakeDirectory(path, logger=None):
if logger is None:
logger = logging.getLogger()
if not NxFiles.isDir(path):
logger.info(str(path) + ' does not exist, attempting to make it.')
NxFiles.makeDirs(path)
def getDepartmentMembers(self, departmentId=1, fetchChild=True, memberStatus=0):
# memberStatus = 0 --> fetch all members
# memberStatus = 1 --> fetch the members followed wechat
# memberStatus = 2 --> fetch disabled members
# memberStatus = 4 --> fetch all the members unfollowed wechat
# https://qyapi.weixin.qq.com/cgi-bin/user/simplelist?access_token=ACCESS_TOKEN&department_id=DEPARTMENT_ID&fetch_child=FETCH_CHILD&status=STATUS
# {
# "errcode": 0,
# "errmsg": "ok",
# "userlist": [
# {
# "userid": "zhangsan",
# "name": "zhangsan",
# "department": [1, 2]
# }
# ]
# }
if memberStatus != 0 and memberStatus != 1 and memberStatus != 2 and memberStatus != 4:
memberStatus = 1
if fetchChild:
myFetchChild = 1
else:
myFetchChild = 0
self.getAccessToken()
for i in xrange(self.retry):
self.sender.resetSession()
self.logger.info('Starting to get the department members ...')
try:
response = self.sender.get('https://qyapi.weixin.qq.com/cgi-bin/user/simplelist?access_token='
+ str(self.accessTokenDict['access_token']) + '&department_id=' +
str(departmentId) + '&fetch_child=' + str(myFetchChild) + '&status=' + str(
memberStatus))
if response.status_code == 200:
responseJson = response.json()
if 'errcode' in responseJson:
if int(responseJson['errcode']) == 40014:
self.logger.debug(
'The errcode from Wechat server is 40014, the errmsg is "invalid access_token", starting to get access_token again.')
NxFiles.removeForce(self.accessTokenFile)
self.accessTokenDict = None
self.getAccessToken()
elif int(responseJson['errcode'] == 0):
self.logger.info('Getting the department members succeeds, the list is:')
self.logger.info(str(responseJson['userlist']))
return responseJson['userlist']
else:
self.logger.info(
'Response errcode is "' + str(responseJson['errcode']) + '", the reason is "' + str(
responseJson['errmsg']) + '" .')
return False
else:
self.logger.info('Json of response has no errcode, try it again.')
else:
self.logger.info('The status code of response from Wechat server is ' + str(
response.status_code) + ', rather than 200!')
self.logger.debug('Response content is:')
self.logger.debug(str(response.content))
self.logger.debug(str('Response json is:'))
self.logger.debug(str(response.json()))
return False
except:
e = sys.exc_info()[0]
self.logger.warning('Getting department members meets exception!')
self.logger.debug(str(e))
time.sleep(1)
def sendText(self, agentId, message, toUserList=None, toPartyList=None, toTagList=None):
# toUserList --> List --> string
# agentId --> integer
# message --> string
# toPartyList --> List --> string
# toTagList --> List --> string
# Request Method: POST
# https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=ACCESS_TOKEN
# {
# "touser": "******", --> Not must
# "toparty": " PartyID1 | PartyID2 ", --> Not must
# "totag": " TagID1 | TagID2 ", --> Not must
# "msgtype": "text", --> Must
# "agentid": 1, --> Must
# "text": { --> Must
# "content": "Holiday Request For Pony(http://xxxxx)" --> Must
# }, --> Must
# "safe":0 --> Not must
# }
# {u'errcode': 0, u'errmsg': u'ok'}
# {u'errcode': 40014, u'errmsg': u'invalid access_token'}
if toUserList:
if not isinstance(toUserList, list):
self.logger.info('The parameter toUserList you pass is not a list!')
return False
if toPartyList:
if not isinstance(toPartyList, list):
self.logger.info('The parameter toPartyList you pass is not a list!')
return False
if toTagList:
if not isinstance(toTagList, list):
self.logger.info('The parameter toTagList you pass is not a list!')
return False
if not toUserList and not toPartyList and not toTagList:
self.logger.info('One of the toUserList, toPartyList, toTagList should be given!')
return False
toUserString = ''
toPartyString = ''
toTagString = ''
if toUserList:
for i in toUserList:
toUserString = toUserString + str(i) + '|'
if toUserString[-1] == '|':
toUserString = toUserString[0:-1]
if toPartyList:
for j in toPartyList:
toPartyString = toPartyString + '|'
if toPartyString[-1] == '|':
toPartyString = toPartyString[0:-1]
if toTagList:
for k in toTagList:
toTagString = toTagString + '|'
if toTagString[-1] == '|':
toTagString = toTagString[0:-1]
if toUserString == '' and toPartyString == '' and toTagString == '':
self.logger.info('One of the toUserList, toPartyList, toTagList should be given!')
return False
self.getAccessToken()
for m in xrange(self.retry):
self.sender.resetSession()
self.sender.addJson('msgtype', 'text')
self.sender.addJson('text', {'content': str(message)})
self.sender.addJson('agentid', int(agentId))
receiver = ''
if toUserString != '':
self.sender.addJson('touser', toUserString)
receiver = receiver + ' ' + toUserString
if toPartyString != '':
self.sender.addJson('toparty', toPartyString)
receiver = receiver + ' ' + toPartyString
if toTagString != '':
self.sender.addJson('totag', toTagString)
receiver = receiver + ' ' + toTagString
try:
self.logger.info(
'Starting to send the message "' + str(message) + '" to "' + receiver + '" via agentid "' + str(
agentId) + '" ...')
response = self.sender.post(
'https://qyapi.weixin.qq.com/cgi-bin/message/send?access_token=' + self.accessTokenDict['access_token'])
if response.status_code == 200:
responseJson = response.json()
if 'errcode' in responseJson:
if int(responseJson['errcode']) == 40014:
self.logger.debug(
'The errcode from Wechat server is 40014, the errmsg is "invalid access_token", starting to get access_token again.')
NxFiles.removeForce(self.accessTokenFile)
self.accessTokenDict = None
self.getAccessToken()
elif int(responseJson['errcode'] == 0):
self.logger.info('Sending the message "' + str(
message) + '" to "' + receiver + '" via agentid "' + str(agentId) + '" succeeds.')
return True
else:
self.logger.info('Sending the message "' + str(
message) + '" to "' + receiver + '" via agentid "' + str(agentId) + '" fails!')
self.logger.info(
'Response errcode is "' + str(responseJson['errcode']) + '", the reason is "' + str(
responseJson['errmsg']) + '" .')
return False
else:
self.logger.info('Json of response has no errcode, try it again.')
else:
self.logger.info('Status code of response from Wechat server is "' + str(
response.status_code) + '" not "200" , please check your parameters!')
self.logger.debug('Response content is:')
self.logger.debug(str(response.content))
self.logger.debug('Response json is:')
self.logger.debug(str(response.json()))
return False
except:
self.logger.warning('Sending the message "' + str(message) + '" to "' + receiver + '" via agentid "' + str(
agentId) + '" meets exception!')
e = sys.exc_info()[0]
self.logger.debug(str(e))
time.sleep(1)
self.logger.info('The sending has been retried ' + str(self.retry) + ' times, but it still fails!')
return False