htcap.py

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

"""
HTCAP - beta 1
Author: filippo.cavallarin@wearesegment.com

This program is free software; you can redistribute it and/or modify it under 
the terms of the GNU General Public License as published by the Free Software 
Foundation; either version 2 of the License, or (at your option) any later 
version.
"""

from __future__ import unicode_literals
import sys
import os
import datetime
import time
import getopt

from core.lib.utils import *
from core.crawl.crawler import Crawler
from core.scan.scanner import Scanner

reload(sys)  
sys.setdefaultencoding('utf8')


def usage():
	infos = get_program_infos()
	print ("htcap ver " + infos['version'] + "\n"
		   "usage: htcap <command>\n" 
		   "Commands: \n"		  
		   "  crawl                  run crawler\n"
		   "  scan                   run scanner\n"		   
		   )	


if __name__ == '__main__':

	if len(sys.argv) < 2:
		usage()
		sys.exit(1)

	elif sys.argv[1] == "crawl":		
		Crawler(sys.argv[2:])
	elif sys.argv[1] == "scan":		
		Scanner(sys.argv[2:])					
	else:
		usage();
		sys.exit(1)

	sys.exit(0)