MycroftAI
/
selene-backend
mirror of https://github.com/MycroftAI/selene-backend.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

added repository method to get the number of devices for an account.

pull/58/head
Chris Veilleux 2019-02-28 21:46:32 -06:00
parent 35fa76e6a2
commit 3c5983a194
2 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
shared/selene/data/device/repository/device.py
View File

@ -31,7 +31,6 @@ class DeviceRepository(object):
def get_devices_by_account_id(self, account_id: str) -> List[Device]: def get_devices_by_account_id(self, account_id: str) -> List[Device]:
"""Fetch all devices associated to a user from a given account id """Fetch all devices associated to a user from a given account id
:param db: psycopg2 connection to mycroft database
:param account_id: uuid :param account_id: uuid
:return: List of User's devices :return: List of User's devices
""" """
@ -42,6 +41,18 @@ class DeviceRepository(object):
sql_results = self.cursor.select_all(query) sql_results = self.cursor.select_all(query)
return [Device(**result) for result in sql_results] return [Device(**result) for result in sql_results]
def get_account_device_count(self, account_id):
query = DatabaseRequest(
sql=get_sql_from_file(
path.join(SQL_DIR, 'get_account_device_count.sql')
),
args=dict(account_id=account_id)
)
sql_results = self.cursor.select_one(query)
return sql_results['device_count']
def get_subscription_type_by_device_id(self, device_id): def get_subscription_type_by_device_id(self, device_id):
"""Return the type of subscription of device's owner """Return the type of subscription of device's owner
:param device_id: device uuid :param device_id: device uuid

1
shared/selene/data/device/repository/sql/get_account_device_count.sql Normal file
View File

@ -0,0 +1 @@
SELECT count(*) AS device_count FROM device.device WHERE account_id = %(account_id)s