manyi/script/spider/tiantian_zone.py

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

import os, sys,time,datetime
import urllib
import requests
from mongoengine import register_connection, PointField, DynamicDocument, StringField
import simplejson as json
import base64
import sys
import json
import base64
import urllib2
from urllib import quote_plus
from urllib2 import urlopen
from urllib2 import Request
from urllib2 import URLError
from urllib import urlencode

from django.db.models.fields import DateTimeField


PROJECT_ROOT = os.path.join(os.path.abspath(os.path.split(os.path.realpath(__file__))[0] + "/.."), '..')
sys.path.insert(0, PROJECT_ROOT)

os.environ.setdefault("DJANGO_SETTINGS_MODULE", "configs.testing")

from script.base import init_env

init_env(interactive = False)

from apps.web.core.db import Searchable

import qrcode

register_connection(alias = 'spider',
                    name = 'spider',
                    host = '116.62.228.194',
                    port = 27017,
                    username = 'dba',
                    password = 'dayuan@2020..',
                    authentication_source = 'admin')


class tiantianPort(Searchable):    
    addr = StringField(default = '')
    portId = StringField(default = '')
    
    meta = {
        'collection': 'tiantian_port',
        'db_alias': 'spider',
        'unique_together': {'portId'}
    }

ports = [port for port in tiantianPort.objects.all()]
zoneDict = {}
for port in ports:
    if not zoneDict.has_key(port.addr):
        zoneDict[port.addr] = port


for zone,port in zoneDict.items():
    try:
        qrCode = 'https://cd1a.cn/?s='+port.portId
        print zone,port.portId,qrCode
        img = qrcode.make(qrCode)
        img.save('F:/tiantian/%s.jpg' % port.portId)
    except Exception,e:
        continue
# 打印文字
print('OK')