From b5a4153288362fbe9336d14822ab73a71209c84c Mon Sep 17 00:00:00 2001
From: Vivek Teega <vivek.teega@gmail.com>
Date: Sat, 8 Dec 2018 14:53:28 +0530
Subject: [PATCH] Basic working code for balance is done

---
 app/__init__.py         |  4 +++-
 app/models.py           | 28 ++++++++++++++++++++++++++++
 app/routes.py           |  8 +++++++-
 app/templates/test.html |  6 ++++++
 config.py               |  5 ++++-
 5 files changed, 48 insertions(+), 3 deletions(-)
 create mode 100644 app/models.py

diff --git a/app/__init__.py b/app/__init__.py
index f48d79f..ecf426c 100644
--- a/app/__init__.py
+++ b/app/__init__.py
@@ -5,6 +5,8 @@ from flask_migrate import Migrate
 
 app = Flask(__name__)
 app.config.from_object(Config)
+db = SQLAlchemy(app)
+migrate = Migrate(app, db)
 
-from app import routes
+from app import routes, models
 
diff --git a/app/models.py b/app/models.py
new file mode 100644
index 0000000..1475652
--- /dev/null
+++ b/app/models.py
@@ -0,0 +1,28 @@
+from app import db
+
+class Transactions(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    address = db.Column(db.String(64), index=True)
+    parentid = db.Column(db.Integer, index=True)
+    transferBalance = db.Column(db.Integer)
+
+    def __repr__(self):
+        return '<User {}>'.format(self.address)
+
+class Transferlogs(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    primaryIDReference = db.Column(db.Integer)
+    transferDescription = db.Column(db.String, index=True)
+    transferIDConsumed = db.Column(db.String, index=True)
+    blockchainReference = db.Column(db.String)
+
+    def __repr__(self):
+        return '<User {}>'.format(self.primaryIDReference)
+
+class Webtable(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    transferDescription = db.Column(db.String, index=True)
+    blockchainReference = db.Column(db.String, index=True, unique=True)
+
+    def __repr__(self):
+        return '<User {}>'.format(self.transferDescription)
\ No newline at end of file
diff --git a/app/routes.py b/app/routes.py
index b6ca6ca..e5f9ef6 100644
--- a/app/routes.py
+++ b/app/routes.py
@@ -2,6 +2,7 @@ from flask import render_template
 from flask import render_template, flash, redirect, url_for
 from app import app
 from app.forms import LoginForm
+import sqlite3
 
 @app.route('/', methods=['GET', 'POST'])
 @app.route('/index', methods=['GET', 'POST'])
@@ -27,6 +28,11 @@ def test():
     form = LoginForm()
     if form.validate_on_submit():
         flash('Balance requested for address {}'.format(form.address.data))
-        return 'string'
+        conn = sqlite3.connect('/home/vivek/Dev/RanchiMall/rmfzeTracking_testnet/tree.db')
+        c = conn.cursor()
+        c.execute("select SUM(transferBalance) from transactiontable WHERE address=='{}'".format(str(form.address.data)))
+        balance = c.fetchall()[0][0]
+        conn.close()
+        return render_template('test.html', form=form, balance=balance)
         # return redirect(url_for('index'))
     return render_template('test.html', form=form)
\ No newline at end of file
diff --git a/app/templates/test.html b/app/templates/test.html
index 12d72ec..6f7a38c 100644
--- a/app/templates/test.html
+++ b/app/templates/test.html
@@ -96,6 +96,12 @@
                                     {% for error in form.address.errors %}
                                         <span style="color: red;">[{{ error }}]</span>
                                     {% endfor %}
+                                    {% if balance %}
+                                        <div class="ml-auto mr-auto">
+                                            <br />
+                                            <p>Balance {{balance}} RMT</p>
+                                        </div>
+                                    {% endif %}
 							    </div>
                             </form>
                         </div>
diff --git a/config.py b/config.py
index c372070..54a3870 100644
--- a/config.py
+++ b/config.py
@@ -2,4 +2,7 @@ import os
 basedir = os.path.abspath(os.path.dirname(__file__))
 
 class Config(object):
-    SECRET_KEY = os.environ.get('SECRET_KEY') or 'you-will-never-guess'
\ No newline at end of file
+    SECRET_KEY = os.environ.get('SECRET_KEY') or 'you-will-never-guess'
+    #SQLALCHEMY_DATABASE_URI = os.environ.get('DATABASE_URL') or os.path.join(basedir, 'tree.db')
+    SQLALCHEMY_DATABASE_URI = os.environ.get('DATABASE_URL') or 'sqlite:///' + os.path.join('/home/vivek/Dev/RanchiMall/rmfzeTracking_testnet', 'app.db')
+    SQLALCHEMY_TRACK_MODIFICATIONS = False
\ No newline at end of file