예제 #1
0
import asyncio
import asyncpg
from gino import Gino
# from api_server.models.url import Url

db = Gino()

# Initialize Queue
db_queue = asyncio.Queue()


async def listen_for_writes(app):
    while True:
        # wait for an item from the producer
        link = await db_queue.get()
        if link is None:
            print(1)
            # the producer emits None to indicate that it is done
            break

        # simulate i/o operation using sleep
        # await asyncio.sleep(2)

        # process the item
        print(f"Received new URL: {link.original_url}")

        await link.create()
예제 #2
0
import asyncio
import traceback

from gino import Gino
from sqlalchemy import ForeignKey

dbo = Gino()


class Server(dbo.Model):
    __tablename__ = 'server'
    guildid = dbo.Column(dbo.Integer(), primary_key=True)  # pylint: disable=no-member


class User(dbo.Model):
    __tablename__ = 'users'
    id = dbo.Column(dbo.Integer(), primary_key=True)  # pylint: disable=no-member
    guild = dbo.Column(dbo.Integer(), None, ForeignKey('Server.guildid'))  # pylint: disable=no-member
    funds = dbo.Column(dbo.Integer())  # pylint: disable=no-member


class dbc:
    async def add_guild(self, id):
        queryRes = Server.query.where(Server.guildid == id)
        #print(queryRes) #THIS BREAKS INITIALIZATION SOMEHOW!
        if (queryRes == None):
            await Server.create(guildid=id)
            return True
        else:
            return False