import os
import sys
runPath = os.path.dirname(os.path.realpath(__file__))
sys.path.append(os.path.join(runPath, "./lib/"))
import pymongo
import redis
from Config import Configuration
# connect to db
db = Configuration.getMongoConnection()
cpe = db.cpe
try:
r = Configuration.getRedisVendorConnection()
except:
sys.exit(1)
for e in cpe.find({}):
try:
value = e['id']
if e['id'].count(':') > 4:
value = ":".join(str(x) for x in (value.split(':')[:5]))
(prefix, cpetype, vendor, product, version) = value.split(':')
except:
pass
r.sadd("prefix:"+prefix, cpetype)
r.sadd("t:"+cpetype, vendor)
r.sadd("v:"+vendor, product)
r.sadd("p:"+product, version)
argparser.add_argument('-v', action='store_true', help='verbose output')
args = argparser.parse_args()
# variables
app = Flask(__name__, static_folder='static', static_url_path='/static')
app.config['MONGO_DBNAME'] = Configuration.getMongoDB()
app.config['SECRET_KEY'] = str(random.getrandbits(256))
pageLength = Configuration.getPageLength()
# login manager
login_manager = LoginManager()
login_manager.init_app(app)
# db connectors
mongo = PyMongo(app)
db = Configuration.getMongoConnection()
redisdb = Configuration.getRedisVendorConnection()
# functions
def matchFilePath(path):
pattern = re.compile('^([a-zA-Z/ 0-9._-])+$')
if pattern.match(path):
return True
else:
return False
def getBrowseList(vendor):
result = {}
if (vendor is None) or type(vendor) == list:
v1 = redisdb.smembers("t:/o")
v2 = redisdb.smembers("t:/a")
v3 = redisdb.smembers("t:/h")
