python检测图片是否存在脚本
使用(此为测试表结构,与公司立场无关),查询结果写入当前目录的1.txt&2.txt中
代码写的烂,大神别吐槽,python刚学2天,不容易
./pycheck404url.py 0 100 1
0:起始便宜量,100 查询总数 1:查询主表
./pycheck404url.py 0 100 2
0:起始便宜量,100 查询总数 2:查询图片表
#!/usr/bin/env python
#coding:utf-8
from mysql import connector
import urllib2
import sys
import socket
HOST = 'localhost'
USER = 'root'
PASSWORD = ''
DATABASE = 'pytest'
ImagePrefix = 'http://image.59store.com'
def pyconnect(**kwargs):
try:
connect = connector.connect(host=kwargs.get('host'), user=kwargs.get('user'), passwd=kwargs.get('password'), db=kwargs.get('database'))
return connect
except Exception as e:
print(e)
return False
def queryData(start=0, limit=100, type=1):
connect = pyconnect(host=HOST, user=USER, password= PASSWORD, database= DATABASE)
if connect != False:
cursor = connect.cursor()
if type == 1:
sql = 'SELECT rid,default_image FROM prefix_repo limit %s,%s'
sql = sql % (start, limit)
else:
sql = 'SELECT rid,url FROM prefix_repoimage limit %s,%s'
sql = sql % (start, limit)
cursor.execute(sql)
data = cursor.fetchall()
return data
else:
return False
def check200(start, limit, type=1):
file = open('%s.txt' % (type),'a')
data = queryData(start, limit, type)
if data != False:
for row in data:
try:
url = ImagePrefix + row[1]
print url
response = urllib2.urlopen(url, timeout=3)
httpCode = response.getcode()
if httpCode != 200:
file.write('%s\t%s \n' % (row[0], url))
else:
print(row[0],'ok')
except urllib2.HTTPError as e:
file.write('%s\t%s \n' % (row[0], url))
except socket.timeout as e:
file.write('%s\t%s \n' % (row[0], url))
file.close()
if __name__ == "__main__":
args = sys.argv
print args
check200(args[1],args[2],int(args[3]))
数据库结构
1.商品主表
CREATE TABLE `prefix_repo` (
`rid` int(10) unsigned NOT NULL AUTO_INCREMENT,
`status` tinyint(3) NOT NULL DEFAULT '0',
`default_image` varchar(100) NOT NULL DEFAULT '',
PRIMARY KEY (`rid`),
KEY `rid` (`rid`,`status`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8;
2.图片表
CREATE TABLE `prefix_repoimage` (
`item_id` int(10) unsigned NOT NULL AUTO_INCREMENT,
`rid` int(10) unsigned NOT NULL,
`url` varchar(120) NOT NULL DEFAULT '',
PRIMARY KEY (`item_id`),
KEY `rid` (`rid`)
) ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=utf8 ROW_FORMAT=COMPACT;
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。