用python写的测试网页和文件编码的小程序

用python写的测试网页和文件编码的小程序 - cloudaice - 博客园

用python写的测试网页和文件编码的小程序

这几天闲着呆在实验室，就自己学学python，不过写出来的程序都是很难看的，将就还是能用吧，在《可爱的python》中学习了cmd模块，于是就照锚画虎的自己琢磨着在其他一些小程序当中也使用上。这是一个测试网页编码或者文件编码的小程序，需要chaedet模块，顺便为了学习使用git，于是就把代码托管到github上了，地址为 git@github.com:cloudaice/codetest.git.

下面贴上代码

# -*- coding:utf-8 -*-
import cmd
import urllib
import sys
import chardet
from chardet.universaldetector import UniversalDetector
class definecode(cmd.Cmd):
    def __init__(self):
        cmd.Cmd.__init__(self)
        self.preaddr="http://"
        self.prompt='(testcode)>'
        self.intro='''
        website 输入网址  输入要检测编码方式的网站的网址
        exit or q    推出程序
'''
    def help_exit(self):
        print "推出程序"
    def do_exit(self,line):
        sys.exit()

    def help_website(self):
        print "输入网站的网址"
    def do_website(self,website):
         if website=='':
             website=raw_input("输入网址 ")
         if not self.preaddr in website:
             website=self.preaddr+website 
         sock = urllib.urlopen(website)
         detector = UniversalDetector()
         for line in sock.readlines():
             detector.feed(line)
             if detector.done:
                 break
         detector.close()
         sock.close()
         result = detector.result
         print "该网站的编码是 "+result['encoding']
     
    def help_files(self):
         print "输入文件完整路径和文件名"
    def do_files(self,filenames):
         if filenames=='':
             filenames=raw_input("输入文件完整路径和文件名")
         f = open(filenames)
         detector = UniversalDetector()
         for line in f.readlines():
             detector.feed(line)
             if detector.done:
                 break
         detector.close()
         f.close()
         result = detector.result
         print "该文件的编码是 "+ result['encoding']
    
    do_q = do_exit

if __name__ == '__main__':
    testcode = definecode()
    testcode.cmdloop()