Tanto ora cambio tutto

database.py

@@ -1,6 +1,6 @@
 import sqlalchemy.exc
-from sqlalchemy import create_engine, Column, Integer, String, Boolean
-from sqlalchemy.orm import sessionmaker
+from sqlalchemy import create_engine, Column, Integer, String, Boolean, DateTime, ForeignKey
+from sqlalchemy.orm import sessionmaker, relationship
 from sqlalchemy.ext.declarative import declarative_base
 import bcrypt
 
@@ -18,6 +18,7 @@ engine = create_engine("sqlite:///db.sqlite")
 Base = declarative_base()
 Session = sessionmaker(bind=engine)
 
+
 class User(Base):
     __tablename__ = "members"
 
@@ -27,6 +28,7 @@ class User(Base):
     royal = Column(Boolean, nullable=False)
     telegram_id = Column(Integer, unique=True)
     discord_id = Column(Integer, unique=True)
+    diario_entries = relationship("Diario")
 
     def __str__(self):
         return self.username
@@ -34,8 +36,21 @@ class User(Base):
     def __repr__(self):
         return f"<User {self.id}: {self.username}>"
 
+
+class Diario(Base):
+    __tablename__ = "diario"
+
+    id = Column(Integer, primary_key=True)
+    text = Column(String, nullable=False)
+    date = Column(DateTime, nullable=False)
+    author = Column(Integer, ForeignKey("members.id"))
+
+    def __repr__(self):
+        return f"<Diario {self.date} {self.text}>"
+
 Base.metadata.create_all(engine)
 
+
 def create_user(username, password, royal=False):
     """Create a new user and add it to the database."""
     # Create a new session
@@ -67,7 +82,8 @@ def change_password(username, newpassword):
 
 
 def login(username, password, enable_exceptions=False):
-    """Try to login using the database password. The session is always returned, while the user object is returned if the login is successful."""
+    """Try to login using the database password.
+    The session is always returned, while the user object is returned if the login is successful."""
     # Create a new session
     session = Session()
     # Find the matching user
@@ -98,8 +114,29 @@ def find_user(username):
     # Return the session and the user
     return session, db_user
 
-session = Session()
-# Generate the database if it's empty
-if session.query(User).first() is None:
-    init_royal_db()
-del session
+
+def migrate_diario():
+    import datetime
+    session = Session()
+    file = open("diario.txt", encoding="utf8")
+    for row in file:
+        entry = row.split("|", 1)
+        new = Diario()
+        new.date = datetime.datetime.fromtimestamp(int(entry[0]))
+        new.text = entry[1]
+        session.add(new)
+    session.commit()
+
+
+def new_diario_entry(dt, text, author):
+    # Create a new session
+    session = Session()
+    # Create a new diario entry
+    entry = Diario()
+    entry.date = dt
+    entry.text = text
+    entry.author = author.id
+    # Add the entry to the database
+    session.add(entry)
+    # Commit the change
+    session.commit()

royalbot.py

@@ -63,21 +63,12 @@ Sintassi: `/diario <frase>`"""
     if len(arguments) == 0:
         await update.message.reply(bot, "⚠ Sintassi del comando non valida.\n`/diario <random | markov | numerofrase>`", parse_mode="Markdown")
         return
-    # Check for non-ASCII characters
-    entry = " ".join(arguments)
-    if not entry.isprintable():
-        await update.message.reply(bot, "⚠ La frase che stai provando ad aggiungere contiene caratteri non ASCII, quindi non è stata aggiunta.\nToglili e riprova!")
-        return
-    # Remove endlines
-    entry = entry.replace("\n", " ")
-    # TODO: check if a end-of-file character can be sent on Telegram
-    # Generate a timestamp
-    time = update.message.date.timestamp()
-    # Write on the diario file
-    file = open("diario.txt", "a", encoding="utf8")
-    file.write(f"{int(time)}|{entry}\n")
-    file.close()
-    del file
+    # Find the user
+    user = currently_logged_in(update)
+    # Prepare the text
+    text = " ".join(arguments).strip()
+    # Add the new entry
+    database.new_diario_entry(update.message.date, text, user)
     # Answer on Telegram
     await update.message.reply(bot, "✅ Aggiunto al diario!")
 
@@ -494,23 +485,17 @@ Sintassi: `!roll <max>`"""
     await bot.send_message(message.channel, f"*Numero generato:* {random.randrange(0, int(arguments[0])) + 1}")
 
 
-async def adduser_telegram(bot, update, arguments):
-    """Aggiungi un utente al database Royal Games!
+async def register_telegram(bot, update, arguments):
+    """Registrati al database Royal Games!
 
-Devi essere un Royal per poter eseguire questo comando.
-
-Sintassi: `/adduser <username> <password>`"""
-    # Check if the user is logged in
-    if not currently_logged_in(update):
-        await update.message.reply(bot, "⚠ Non hai ancora eseguito l'accesso! Usa `/sync`.", parse_mode="Markdown")
+Sintassi: `/register <username> <password>`"""
+    # One account per user only!
+    if currently_logged_in(update) is not None:
+        await update.message.reply(bot, "⚠ Sei già registrato!")
         return
-    # Check if the currently logged in user is a Royal Games member
-    if not currently_logged_in(update).royal:
-        await update.message.reply(bot, "⚠ Non sei autorizzato a eseguire questo comando.")
-        return
-    # Check the command syntax
-    if len(arguments) != 2:
-        await update.message.reply(bot, "⚠ Sintassi del comando non valida.\n`/adduser <username> <password>`", parse_mode="Markdown")
+    # Check if the username is printable
+    if not arguments[0].isprintable():
+        await update.message.reply(bot, "⚠ Username non valido.")
         return
     # Try to create a new user
     try:
@@ -519,8 +504,11 @@ Sintassi: `/adduser <username> <password>`"""
         await update.message.reply(bot, "⚠ Qualcosa è andato storto nella creazione dell'utente. Per altre info, guarda i log del bot.")
         raise
     else:
-        await update.message.reply(bot, "✅ Creazione riuscita!")
-
+        session, logged_user = database.login(arguments[0], arguments[1])
+        logged_user.telegram_id = update.message.sent_from.user_id
+        session.commit()
+        print(f"{logged_user} ha sincronizzato l'account di Telegram.")
+        await update.message.reply(bot, f"Sincronizzazione riuscita!\nSei loggato come `{logged_user}`.", parse_mode="Markdown")
 
 async def toggleroyal_telegram(bot, update, arguments):
     """Inverti lo stato di Royal di un utente.
@@ -568,7 +556,7 @@ if __name__ == "__main__":
     b.commands["markov"] = markov_telegram
     b.commands["cv"] = cv_telegram
     b.commands["roll"] = roll_telegram
-    b.commands["adduser"] = adduser_telegram
+    b.commands["register"] = register_telegram
     b.commands["toggleroyal"] = toggleroyal_telegram
     # Init Discord bot commands
     d.commands["sync"] = sync_discord