dockerized the application

Dockerfile

@@ -0,0 +1,23 @@
+FROM python:3.9-slim-buster
+
+WORKDIR /app
+
+COPY requirements.txt .
+
+COPY .gf /root/.gf
+
+RUN apt-get update -y && apt-get install -y --no-install-recommends gcc libcurl4-openssl-dev libc6-dev libssl-dev dnsutils && rm -rf /var/lib/apt/lists/*
+
+RUN pip install --upgrade pip
+
+RUN pip install -r requirements.txt
+
+COPY app.py .
+
+COPY . .
+
+ENV PYTHONWARNINGS=ignore
+
+RUN chmod -R 777 .
+
+CMD ["python3", "app.py"]

Dockerfile.bak

@@ -0,0 +1,16 @@
+FROM python:3.9
+
+WORKDIR /app
+
+COPY . /app
+
+RUN apt-get update && apt-get install -y libcurl4-openssl-dev 
+
+RUN pip install --upgrade pip
+
+RUN pip install --no-cache-dir -r requirements.txt
+
+EXPOSE 5000
+
+CMD ["python", "app.py"]
+

app.py

@@ -5,6 +5,15 @@
 
 app = Flask(__name__)
 
+conn = psycopg2.connect(
+    host="db",
+    database="test_db",
+    user="test_user",
+    password="test_password"
+)
+
+cur = conn.cursor()
+
 #########################################################################
 
 @app.route("/")
@@ -21,17 +30,10 @@ def home():
 def adder():
  if request.method == "POST":
   url = request.form.get('url')
-  DATABASE_URL = os.environ['DATABASE_URL']
-
-  conn = psycopg2.connect(DATABASE_URL, sslmode='require')
-  cur = conn.cursor()
 
   cur.execute(f"insert into queue (target) values ('{url}')")
   conn.commit()
 
-  cur.close()
-  conn.close()
-
   return render_template("index.html", info="Added to queue !")
 
 #########################################################################
@@ -369,9 +371,6 @@ def check_queue():
   res = ''
   res = res + q1
 
-  DATABASE_URL = os.environ['DATABASE_URL']
-  conn = psycopg2.connect(DATABASE_URL, sslmode='require')
-  cur = conn.cursor()
   cur.execute("select * from queue")
   t = cur.fetchall()
 
@@ -387,8 +386,6 @@ def check_queue():
 
   res = res + q2
   conn.commit()
-  cur.close()
-  conn.close()
 
   return res
 
@@ -724,9 +721,6 @@ def scanned():
   res = ''
   res = res + s1
 
-  DATABASE_URL = os.environ['DATABASE_URL']
-  conn = psycopg2.connect(DATABASE_URL, sslmode='require')
-  cur = conn.cursor()
   cur.execute("select domain from output")
   t = cur.fetchall()
 
@@ -742,8 +736,6 @@ def scanned():
 
   res = res + s2
   conn.commit()
-  cur.close()
-  conn.close()
   return res
 
 #########################################################################
@@ -755,13 +747,9 @@ def delete_item():
   key = list(dic.keys())
   id = key[0]
 
-  DATABASE_URL = os.environ['DATABASE_URL']
-  conn = psycopg2.connect(DATABASE_URL, sslmode='require')
-  cur = conn.cursor()
   cur.execute("delete from queue where id={}".format(id))
   conn.commit()
-  cur.close()
-  conn.close()
+
   return redirect(url_for("check_queue"))
 
 #########################################################################
@@ -781,9 +769,6 @@ def about():
 @app.route("/output/<url>")
 def output(url):
   url = url
-  DATABASE_URL = os.environ['DATABASE_URL']
-  conn = psycopg2.connect(DATABASE_URL, sslmode='require')
-  cur = conn.cursor()
 
   cur.execute(f"select result from output where domain = '{url}'")
   t = cur.fetchall()
@@ -797,8 +782,7 @@ def output(url):
   f.close()
 
   conn.commit()
-  cur.close()
-  conn.close()
+
   return send_file("/app/results/{}-output.txt".format(url))
 
 #########################################################################
@@ -822,9 +806,6 @@ def results():
 @app.route("/gau/<url>")
 def gau_urls(url):
   url = url
-  DATABASE_URL = os.environ['DATABASE_URL']
-  conn = psycopg2.connect(DATABASE_URL, sslmode='require')
-  cur = conn.cursor()
 
   cur.execute(f"select gau from output where domain = '{url}'")
   t = cur.fetchall()
@@ -838,11 +819,11 @@ def gau_urls(url):
   f.close()
 
   conn.commit()
-  cur.close()
-  conn.close()
+
   return send_file("/app/results/{}-gau.txt".format(url))
 
 #########################################################################
 
 if __name__ == "__main__":
-  app.run(port=8000)
+  app.run(host='0.0.0.0', port=5000)
+

cron.py

@@ -3,9 +3,13 @@
 import psycopg2
 import base64
 
-DATABASE_URL = os.environ['DATABASE_URL']
+conn = psycopg2.connect(
+    host="db",
+    database="test_db",
+    user="test_user",
+    password="test_password"
+)
 
-conn = psycopg2.connect(DATABASE_URL, sslmode='require')
 cur = conn.cursor()
 
 #checks the db queue table every 60 secs

docker-compose.yml

@@ -0,0 +1,29 @@
+version: '3'
+
+services:
+  db:
+    image: postgres
+    container_name: reconator-db
+    environment:
+      POSTGRES_USER: test_user
+      POSTGRES_PASSWORD: test_password
+      POSTGRES_DB: test_db
+    volumes:
+      - db-data:/var/lib/postgresql/data
+    ports:
+      - "5432:5432"
+
+  app:
+    build: .
+    container_name: reconator-app
+    environment:
+      CHAT_ID: 1214546863
+      API_KEY: 6247297064:AAFlsbd24qvsVBJXwwoBwbf562F6ty8ml-M
+    ports:
+      - "5000:5000"
+    depends_on:
+      - db
+
+volumes:
+  db-data:
+

docker-compose.yml.bak

@@ -0,0 +1,29 @@
+version: '3'
+
+services:
+  db:
+    image: postgres
+    restart: always
+    environment:
+      POSTGRES_USER: myuser
+      POSTGRES_PASSWORD: mypassword
+      POSTGRES_DB: mydb
+    volumes:
+      - db-data:/var/lib/postgresql/data
+      - ./pg_hba.conf:/var/lib/postgresql/data/pg_hba.conf
+    ports:
+      - 5432:5432
+
+  app:
+    build: .
+    ports:
+      - 5000:5000
+    environment:
+      CHAT_ID: 1214546863
+      API_KEY: 6247297064:AAFlsbd24qvsVBJXwwoBwbf562F6ty8ml-M
+    depends_on:
+      - db
+
+volumes:
+  db-data:
+

initialise.py

@@ -10,11 +10,16 @@
 
 CHAT_ID = os.environ['CHAT_ID']
 API_KEY = os.environ['API_KEY']
-DATABASE_URL = os.environ['DATABASE_URL']
 
 bot = telebot.TeleBot(API_KEY)
 
-conn = psycopg2.connect(DATABASE_URL, sslmode='require')
+conn = psycopg2.connect(
+    host="db",
+    database="test_db",
+    user="test_user",
+    password="test_password"
+)
+
 cur = conn.cursor()
 
 # to save the result of recon on target

insert.py

@@ -11,15 +11,20 @@
 last = False
 
 url = sys.argv[1]
-DATABASE_URL = os.environ['DATABASE_URL']
 
 try:
   temp = sys.argv[2]
   last = True
 except:
   pass
 
-conn = psycopg2.connect(DATABASE_URL, sslmode='require')
+conn = psycopg2.connect(
+    host="db",
+    database="test_db",
+    user="test_user",
+    password="test_password"
+)
+
 cur = conn.cursor()
 
 #checking if any previous entry in the database to avoid duplicates