def clear_db():
"""Wipes the database
"""
from pymongo import MongoClient
client = MongoClient(MONGO_HOST, 27017)
db = client.eve
db.drop_collection('nodes')
db.drop_collection('node_types')
db.drop_collection('tokens')
db.drop_collection('users')
def clear_db():
"""Wipes the database
"""
from pymongo import MongoClient
client = MongoClient(MONGO_HOST, 27017)
db = client.eve
db.drop_collection('nodes')
db.drop_collection('node_types')
db.drop_collection('tokens')
db.drop_collection('users')
record = LuckyNumber(winning_number_record)
if save_to_db:
get_or_create(record.toJSON(), numbers_collection)
if noisy:
print record
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("-t", "--test", help="Run the doctests", action="store_true")
parser.add_argument("-s", "--scrape", help="Scrape the lottery website", action="store_true")
parser.add_argument(
"-p",
"--print_data",
help="Print the scraped data in a csv format so that you can do python lucky.py -s -p > data.csv",
action="store_true",
)
parser.add_argument("-sd", "--save_data", help="Save the scraped data to the MongoDB database", action="store_true")
parser.add_argument(
"-ddb", "--delete_db", help="Save the scraped data to the MongoDB database", action="store_true"
)
args = parser.parse_args()
if args.test:
doctest.testmod()
if args.scrape:
main_scrape(noisy=args.print_data, save_to_db=args.save_data)
if args.delete_db:
db.drop_collection(numbers_collection)
def tearDown(self):
db.drop_collection('log')
