#3 Lagrar nu koordinater för kartor i mongodb🥬

README.md

@@ -30,7 +30,7 @@ Robot för automatisering av lagerarbete m.m 🤖
   - Gör en pull request
   noob
 
-### Göra ändringar i koden för visualering:
+### Göra ändringar i koden för visualering
  - Innan virtuella miljön avaktiveras
    - Navigera till Gymnasiearbete
    - ```pip freeze > requirements.txt```

requirements.txt

@@ -4,6 +4,7 @@ kiwisolver==1.2.0
 matplotlib==3.3.2
 numpy==1.19.2
 Pillow==8.0.1
+pymongo==3.11.0
 pyparsing==2.4.7
 python-dateutil==2.8.1
 six==1.15.0

visualisering/databas.py

@@ -0,0 +1,81 @@
+import pymongo
+from datetime import datetime
+
+class Databas:
+    '''
+    Klass för att lagra och hämta kartor.
+    '''
+    def __init__(self):
+        '''
+        Försöker ansluta till databasen.
+        '''
+        try:
+            self.handtag = pymongo.MongoClient("mongodb://localhost:27017/")
+            self.db = self.handtag["kartor"]
+            self.karta = self.välj()
+        except Exception as e:
+            print('Starta mongo!')
+
+    def kartor(self) -> list:
+        '''
+        Hämtar namnet på kartorna som redan finns.
+        '''
+        return self.db.list_collection_names()
+
+    def kista(self, villkor: dict = {}) -> list:
+        '''
+        Öppnar upp kistan och returnerar dess koordinater.
+        '''
+        if self.karta:
+            return self.karta.find({}, villkor)
+
+    def välj(self):
+        '''
+        Välj om en ny karta ska skapas eller om en befintlig karta ska användas.
+        '''
+        kartor = self.kartor()
+        antal = len(kartor)
+
+        if antal > 0:
+            print('Dessa kartor fanns redan i databasen\n')
+            for i in range(antal):
+                print('[{}]: {}'.format(i, kartor[i]))
+
+            try:
+                skapa = int(input('Vill du använda en befintlig karta? (0/1) '))
+            except Exception:
+                return self.ny()
+
+            if not skapa:
+                return self.ny()
+
+            try:
+                index = int(input('Vilken karta vill du använda? (0-{}) '.format(antal-1)))
+                karta = kartor[index]
+            except Exception:
+                print('Felaktig inmatning')
+
+            return self.db[karta]
+
+        else:
+            return self.ny()
+
+    def ny(self):
+        '''
+        Skapar en ny karta.
+        '''
+        print('Gör en ny karta')
+        tid = datetime.now()
+        namn = tid.strftime("%Y-%m-%d %H:%M:%S")
+        return self.db[namn]
+
+    def spara(self, koordinater: list):
+        '''
+        Sparar en bunt koordinater som hör ihop med en viss karta i databasen.
+        '''
+        if self.karta:
+            return self.karta.insert_many(koordinater)
+        else:
+            print('Ingen karta har valts')
+
+

visualisering/karta.py

@@ -2,6 +2,7 @@
 import matplotlib.pyplot as plt 
 import matplotlib.animation as anim
 from koordinat import Koordinat
+from databas import Databas
 from math import cos, sin, pi
 
 class Karta:
@@ -13,13 +14,15 @@ def __init__(self):
         Initierar kartan.
         '''
         self.koordinater = list()
+        self.db = Databas()
+
         self.fig = plt.figure()
 
         self.ax = self.fig.gca()
         self.ax.set_ylabel('cm')
         self.ax.set_xlabel('cm')
 
-        self.position = (0, 0)
+        self.position = (0, 0) # Robotens startpostion markeras från origo
 
         plt.grid() # Rutnät över grafen
 
@@ -35,8 +38,12 @@ def läs(self, kö: queue.Queue):
                 self.position = (koord.x, koord.y)
 
             vägg = self.konvertera(koord)
-            self.koordinater.append(vägg)    
-
+            self.koordinater.append(vägg)
+
+            if len(self.koordinater) == 180:
+                self.db.spara(self.koordinater) 
+                self.koordinater.clear()
+
             kö.task_done()
 
     def konvertera(self, koordinat: Koordinat) -> tuple:
@@ -46,7 +53,7 @@ def konvertera(self, koordinat: Koordinat) -> tuple:
         x = koordinat.distans * cos(pi/2 - koordinat.vinkel) + koordinat.x
         y = koordinat.distans * sin(pi/2 - koordinat.vinkel) + koordinat.y
 
-        return (x, y)
+        return {'x': x, 'y': y, 'stegX': koordinat.x, 'stegY': koordinat.y} # Informationen som lagras i databasen.
 
     def visa(self):
         '''
@@ -59,9 +66,13 @@ def uppdatera(self, i):
         ''' 
         Uppdatera kartan med nya värden.
         '''
-        if self.koordinater:
-            x = [kord[0] for kord in self.koordinater]
-            y = [kord[1] for kord in self.koordinater]
-
-            plt.scatter(x, y) # Markera väggarnas position med en prick
-            plt.plot(self.position, '*') # Markera vart roboten är i koordinatsystemet
+        koordinater = self.db.kista({'_id': 0, 'x': 1, 'y': 1})
+
+        x, y = [], []
+
+        for koordinat in koordinater:
+            x.append(koordinat['x'])
+            y.append(koordinat['y'])
+
+        plt.scatter(x, y) # Markera väggarnas position med en prick
+        plt.plot(self.position, '*') # Markera vart roboten är i koordinatsystemet