def add_user_in_db(update, session):
user = update.message.from_user
username = getattr(user, 'username', sqlalchemy.null())
session.add(User(telegram_username=username, telegram_user_id=user.id,
first_name=user.first_name, last_name=user.last_name,
date_registration=datetime.datetime.now(), last_activity=datetime.datetime.now()))
session.commit()
def create_server(dser):
server_sql = q(Server).filter_by(id=str(dser.id)).first()
if server_sql:
print("Server already exists in the database")
return None
new_server = Server(id=str(dser.id))
session.add(new_server)
session.commit()
return new_server
def create_member(dmem):
user_sql = q(User).filter_by(id=str(dmem.id)).first()
if user_sql:
print("Use already exists in database")
return None
new_user = User(id=str(dmem.id))
session.add(new_user)
session.commit()
return new_user
def create_membership(dmem, dser):
membership_sql = q(Membership).filter_by(user_id=str(dmem.id),
server_id=str(dser.id)).first()
if membership_sql:
print("Membership already exists in the database")
return None
new_membership = Membership(user_id=str(dmem.id), server_id=str(dser.id))
session.add(new_membership)
session.commit()
return new_membership
async def shorturl(self, ctx, url: str):
"""Shorten the given URL, the returned URL will redirect to the given URL."""
if len(url) > 2000:
raise URLTooLarge()
url_split = url.split(".")
if not url.startswith("http://") and not url.startswith("https://"):
url = "http://" + url
if len(url_split) > 1:
short = get_unique_short()
new_shorturl = ShortUrl(location=url, short=short)
session.add(new_shorturl)
session.commit()
await ctx.send(
f":white_check_mark: URL shortened to https://www.oakchair.xyz/s/{short}"
)
else:
raise InvalidUrl()
def add_user(user_id):
new_user = User(id=user_id, balance=0, level=1, xp=0)
session.add(new_user)
session.commit()
def add_server(guild_id):
new_server = Server(id=guild_id)
session.add(new_server)
session.commit()