#!/usr/bin/env python
# coding:utf-8

# Author: phan

import os
import yaml
import commands
import time
import codecs

# 配置区域
# 日志路径
LOGFILE = '/var/log/pyrsync.log'
# 是否开启日志
ENABLE_LOG = True
# 啰嗦模式
VERBOSE_MODE = True

# 自定义常量
BASEDIR = os.path.dirname(__file__)

# 获取目录文件列表
def get_file_lists(dirpath):
    filelists = []
    for filename in os.listdir(dirpath):
        filelists.append(os.path.join(dirpath,filename))
    return filelists

# 获取当前时间
def get_time():
    return time.strftime('%Y-%m-%d %H:%M:%S ',time.localtime(time.time()))

# 记录日志
def logger(line):
    if ENABLE_LOG:
        if os.path.isfile(LOGFILE):
            with codecs.open(LOGFILE,'a','utf-8') as f:
                f.write("{time}{logline}\n".format(time=get_time(), logline=line))
        else:
            with codecs.open(LOGFILE,'w','utf-8') as f:
                f.write("{time}{logline}\n".format(time=get_time(), logline=line))

# 记录命令日志
def logger_commands(status,result):
    logger('result:{status}'.format(status=status))
    if VERBOSE_MODE:
        logger(result)

# 执行命令
def runcommand(commandline):
    return commands.getstatusoutput(commandline)

# 部署场景
def deploy_sences(isLocal,shellfile,actor=''):
    commandline = os.path.join(BASEDIR, 'script', shellfile)
    if isLocal:
        (status,result) = runcommand(commandline)
        logger_commands(status,result)
    else:
        upload_script = 'scp {command} {name}@{ip}:/tmp/pyrsync_{scriptName}'.format(
            command=commandline,
            name=actor['name'],
            ip=actor['ip'],
            scriptName=os.path.basename(commandline)
        )
        (status,result) = runcommand(upload_script)
        logger_commands(status, result)
        commandline = 'ssh {name}@{ip} "chmod u+x /tmp/pyrsync_{command} && /tmp/pyrsync_{command} && rm -f /tmp/pyrsync_{command}"'.format(
            name=actor['name'],
            ip=actor['ip'],
            command=os.path.basename(commandline),
        )
        (status,result) = runcommand(commandline)
        logger_commands(status, result)

# 同步文件
def sync_files(actor,sence):
    rsyncCommand = 'rsync {action} --exclude-from="{exclude}" {name}@{ip}:{path} {savepath}'.format(
        action=actor['action'],
        exclude=os.path.join(BASEDIR,'exclude',sence['exclude']),
        name=actor['name'],
        ip=actor['ip'],
        port=actor['port'],
        path=sence['item'],
        savepath=sence['savepath']
    )
    logger('Command:{command}'.format(command=rsyncCommand))
    (status,result) = runcommand(rsyncCommand)
    logger_commands(status,result)

# 执行剧本
def play_playbooks(filelists):
    for filelist in filelists:
        if os.path.isfile(filelist) and os.path.splitext(filelist)[-1]=='.yaml':
            with open(filelist,'r') as f:
                playbook = yaml.load(f)
                sences = playbook['playbook']['sences']
                for sence in sences:
                    if playbook['execute']['pre-local']:
                        deploy_sences(True,playbook['execute']['pre-local'])
                    if playbook['execute']['pre-remote']:
                        deploy_sences(False, playbook['execute']['pre-remote'],playbook['actor'])
                    sync_files(playbook['actor'],sence)
                    if playbook['execute']['post-remote']:
                        deploy_sences(False,playbook['execute']['post-remote'],playbook['actor'])
                    if playbook['execute']['post-local']:
                        deploy_sences(True,playbook['execute']['post-local'])

if __name__=='__main__':
    play_playbooks(get_file_lists(os.path.join(BASEDIR,'playbook')))

#!/usr/bin/env python
#coding:utf-8

from random import randint

"""
    词的频次统计
    统计重复的词的次数，用字典的形式展现
"""

data = [randint(0,10) for _ in xrange(0,20)]

print data

## 常规处理
# 根据键生成一个字典，赋值0
ddata = dict.fromkeys(data,0)

print ddata
# 迭代统计
for x in data:
    ddata[x]+=1

print ddata

## 用collections 的counter

from collections import Counter

ddata2 = Counter(data)

# 统计最高的4位
print ddata2.most_common(4)

#!/usr/bin/env python
#coding:utf-8

from random import randint

"""
    列表筛选
    生成一个有20个随机数的数组 data
    从 data 列表中筛选出大于0的数字
"""

data = [randint(-100,100) for _ in xrange(10)]

print data

## 迭代
def test1(d):
    s = []
    for x in data:
        if x >=0:
            s.append(x)
    return s

## 用filter
def test2(d):
    return filter(lambda x:x>=0,d)

## 用列表解析
def test3(d):
    return [x for x in data if x>=0]

print test1(data)
print test2(data)
print test3(data)

##  迭代最慢、列表解析最快

"""
    字典的筛选
    生成键值在56-100之间的随机数10个
    找出大于90的数
"""

data = {x:randint(56,100) for x in xrange(1,11)}
print data

def test4(d):
    return {k:v for k,v in d.iteritems() if v>=90}

print test4(data)

import socket
#import codecs

#encoder = codecs.getencoder("utf-8")
senddata = '{"action":"compute","device":"ios","map":"0","start":"2017-05-19","day":"25","type":"17","orientation":"0","order_total_id":"950","store_id":"12"}EOF\n'

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect(('dev.xxxx.net', 31))
s.send(senddata)

buffer = []
while True:
    d = s.recv(65535)
    if d:
        buffer.append(d)
    else:
        break
    if 'EOF' in d[-5:-1]:
        break
# while True:
#     d = s.recv(1024)
#     if d:
#         buffer.append(d)
#     else:
#         break
data = ''.join(buffer)
print data
s.close()

#!/bin/env python
#coding:utf-8
from collections import deque
import datetime

# 递归版
def fibonacci(num):
    if num<=2:
        return 1
    else:
        return fibonacci(num-1)+fibonacci(num-2)

# 迭代版
def fibonacci2(num):
    s=deque()
    result=0
    for i in range(1,num+1):
        if i<=2:
            result=1
            s.append(result)
        else:
            result=s[0]+s[1]
            s.append(result)
            s.popleft()
    return result


n=40
print(datetime.datetime.now())
print('fibonacci(%d)=%d' % (n, fibonacci(n)))
print(datetime.datetime.now())
print('fibonacci2(%d)=%d' % (n, fibonacci2(n)))
print(datetime.datetime.now())