import datetime
from flask import Flask
from flask_superadmin import Admin, model,ModelAdmin
#from flask_superadmin.model.backends.mongoengine import ModelAdmin

try:
    from mongoengine import *
except ImportError:
    exit('You must have mongoengine installed. Install it with the command:\n\t$> easy_install mongoengine')




# Create application
app = Flask(__name__)

# Create dummy secret key so we can use sessions
app.config['SECRET_KEY'] = '123456790'
#connect('rdps', host='mongodb://localhost/rdps')

mongodb_settings = {
    'db':'rdps',
    'host':'localhost',
    'port':27017}



connect(**mongodb_settings)

class PHONE(EmbeddedDocument):
    active = BooleanField(label = "Status", default=True)
    number =StringField(label = "Phone Number")
    created_at = DateTimeField(label = "Created At", default=datetime.datetime.now)



class CC(EmbeddedDocument):
    fname = StringField()
    lname = StringField()
    street = StringField()
    city = StringField()
    state = StringField()
    zipcode = IntField()
    active = BooleanField(default=True)
    created_at = DateTimeField(default=datetime.datetime.now)
    cc_number = StringField()
    exp_month = StringField()
    exp_year = StringField()



class RDP(Document):
    label = StringField()
    ip_address = StringField()
    username = StringField(default='Administrator')
    key = StringField()
    password = StringField()
    decrypted_password = StringField()
    created_by = StringField()
    status = StringField()
    state = StringField()
    instance_id = StringField()
    raw_dict = DynamicField()
    phone_number = ListField(EmbeddedDocumentListField(PHONE))
    credit_cards = ListField(EmbeddedDocumentField(CC))





# Defining MongoEngine Documents


# Flask views
@app.route('/')
def index():
    return '<a href="/admin/">Click me to get to Admin!</a>'





if __name__ == '__main__':
    #db = MongoEngine(app)
    admin = Admin(app, 'Simple Models')
    class phoneModel(ModelAdmin):
        list_display = ('number','active','created_at')
        fields = ('number','active','created_at')


    class creditModel(ModelAdmin):
        list_display = ('cc_number','created_at','fname','lname','street','city','state','active')
        fields = ('cc_number','created_at','fname','lname','street','city','state','active')



    class rdpModel(ModelAdmin):
        list_display = ('label','ip_address','decrypted_password','username','instance_id','credit_cards','phone_number','status')
        fields = ('label','ip_address','decrypted_password','username','instance_id','credit_cards','phone_number','status')
    admin = Admin(app, 'Simple Models')
    admin.register(RDP,rdpModel)
    admin.register(cc,creditModel)
    admin.register(phone,phoneModel)



    #db.create_all()
    # Start app
    app.debug = True
    app.run('0.0.0.0', 8000)