cms.py

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

"""
Function: CMS识别系统
Author:   w8ay
Time:     2016年10月21日 16/17/33
"""

import requests
import hashlib
import os

data = []


# socket.setdefaulttimeout(10)

def get_md5_value(src):
    myMd5 = hashlib.md5()
    myMd5.update(src)
    myMd5_Digest = myMd5.hexdigest()
    return myMd5_Digest

#
# def getmd5(url):
#     src = requests.get(url).content
#     md5 = get_md5_value(src)
#     return md5


def init():
    file_url = os.path.abspath(os.path.split(os.path.realpath(__file__))[0] + r"/config/dna.txt")
    file = open(file_url)
    try:
        for line in file:
            str = line.strip().split(" ")
            ls_data = {}
            if len(str) == 3:
                ls_data["url"] = str[0]
                ls_data["name"] = str[1]
                ls_data["md5"] = str[2]
                data.append(ls_data)
    finally:
        file.close()


def cms(url):
    if url is None:
        print "url is Null!"
        return
    url = url.rstrip("/")
    init()
    for dataline in data:
        _url = url + dataline["url"]
        try:
            status = requests.head(_url, timeout=10).status_code
        except Exception, e:
            msg = str(e)


        if status == 200:
            md5 = get_md5_value(requests.get(_url).content)
            # print md5
            if md5 == dataline["md5"]:
                dataline["url"] = _url
                return dataline
    msg = {"error": "没有找到cms"}
    return msg