#11916

pending completion

Build # #11916

Build Type

push

travis-ci

Committed by

web-flow

Commit Message

<a href="https://github.com/datajoint/datajoint-python/commit/<a class=hub.com/datajoint/datajoint-python/commit/cf4bd8ad25fc7a3bf08299ee6392fa1043bb3ed3">cf4bd8ad2<a href="https://github.com/datajoint/datajoint-python/commit/cf4bd8ad25fc7a3bf08299ee6392fa1043bb3ed3">">Merge </a><a class="double-link" href="https://github.com/datajoint/datajoint-python/commit/<a class="double-link" href="https://github.com/datajoint/datajoint-python/commit/77d6fa11d9ba7215a38327ed59bc41e96b1d7cdd">77d6fa11d</a>">77d6fa11d</a><a href="https://github.com/datajoint/datajoint-python/commit/cf4bd8ad25fc7a3bf08299ee6392fa1043bb3ed3"> into 413e2f90d">413e2f90d</a>

Run Details

797 of 797 new or added lines in 25 files covered. (100.0%)

2988 of 3313 relevant lines covered (90.19%)

0.9 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

40.0

/datajoint/admin.py

import pymysql
from getpass import getpass
from .connection import conn
from .settings import config
from .utils import user_choice


def set_password(
    new_password=None, connection=None, update_config=None
):  # pragma: no cover
    connection = conn() if connection is None else connection
    if new_password is None:
        new_password = getpass("New password: ")
        confirm_password = getpass("Confirm password: ")
        if new_password != confirm_password:
            print("Failed to confirm the password! Aborting password change.")
            return
    connection.query("SET PASSWORD = PASSWORD('%s')" % new_password)
    print("Password updated.")

    if update_config or (
        update_config is None and user_choice("Update local setting?") == "yes"
    ):
        config["database.password"] = new_password
        config.save_local(verbose=True)


def kill(restriction=None, connection=None, order_by=None):  # pragma: no cover
    """
    view and kill database connections.
    :param restriction: restriction to be applied to processlist
    :param connection: a datajoint.Connection object. Default calls datajoint.conn()
    :param order_by: order by a single attribute or the list of attributes. defaults to 'id'.

    Restrictions are specified as strings and can involve any of the attributes of
    information_schema.processlist: ID, USER, HOST, DB, COMMAND, TIME, STATE, INFO.

    Examples:
        dj.kill('HOST LIKE "%compute%"') lists only connections from hosts containing "compute".
        dj.kill('TIME > 600') lists only connections in their current state for more than 10 minutes
    """

    if connection is None:
        connection = conn()

    if order_by is not None and not isinstance(order_by, str):
        order_by = ",".join(order_by)

    query = (
        "SELECT * FROM information_schema.processlist WHERE id <> CONNECTION_ID()"
        + ("" if restriction is None else " AND (%s)" % restriction)
        + (" ORDER BY %s" % (order_by or "id"))
    )

    while True:
        print("  ID USER         HOST          STATE         TIME    INFO")
        print("+--+ +----------+ +-----------+ +-----------+ +-----+")
        cur = (
            {k.lower(): v for k, v in elem.items()}
            for elem in connection.query(query, as_dict=True)
        )
        for process in cur:
            try:
                print(
                    "{id:>4d} {user:<12s} {host:<12s} {state:<12s} {time:>7d}  {info}".format(
                        **process
                    )
                )
            except TypeError:
                print(process)
        response = input('process to kill or "q" to quit > ')
        if response == "q":
            break
        if response:
            try:
                pid = int(response)
            except ValueError:
                pass  # ignore non-numeric input
            else:
                try:
                    connection.query("kill %d" % pid)
                except pymysql.err.InternalError:
                    print("Process not found")


def kill_quick(restriction=None, connection=None):
    """
    Kill database connections without prompting. Returns number of terminated connections.
    :param restriction: restriction to be applied to processlist
    :param connection: a datajoint.Connection object. Default calls datajoint.conn()

    Restrictions are specified as strings and can involve any of the attributes of
    information_schema.processlist: ID, USER, HOST, DB, COMMAND, TIME, STATE, INFO.

    Examples:
        dj.kill('HOST LIKE "%compute%"') terminates connections from hosts containing "compute".
    """
    if connection is None:
        connection = conn()

    query = (
        "SELECT * FROM information_schema.processlist WHERE id <> CONNECTION_ID()"
        + ("" if restriction is None else " AND (%s)" % restriction)
    )

    cur = (
        {k.lower(): v for k, v in elem.items()}
        for elem in connection.query(query, as_dict=True)
    )
    nkill = 0
    for process in cur:
        connection.query("kill %d" % process["id"])
        nkill += 1
    return nkill

1	import pymysql	1✔
2	from getpass import getpass	1✔
3	from .connection import conn	1✔
4	from .settings import config	1✔
5	from .utils import user_choice	1✔
6
7
8	def set_password(
9	new_password=None, connection=None, update_config=None
10	): # pragma: no cover
11	connection = conn() if connection is None else connection
12	if new_password is None:
13	new_password = getpass("New password: ")
14	confirm_password = getpass("Confirm password: ")
15	if new_password != confirm_password:
16	print("Failed to confirm the password! Aborting password change.")
17	return
18	connection.query("SET PASSWORD = PASSWORD('%s')" % new_password)
19	print("Password updated.")
20
21	if update_config or (
22	update_config is None and user_choice("Update local setting?") == "yes"
23	):
24	config["database.password"] = new_password
25	config.save_local(verbose=True)
26
27
28	def kill(restriction=None, connection=None, order_by=None): # pragma: no cover
29	"""
30	view and kill database connections.
31	:param restriction: restriction to be applied to processlist
32	:param connection: a datajoint.Connection object. Default calls datajoint.conn()
33	:param order_by: order by a single attribute or the list of attributes. defaults to 'id'.
34
35	Restrictions are specified as strings and can involve any of the attributes of
36	information_schema.processlist: ID, USER, HOST, DB, COMMAND, TIME, STATE, INFO.
37
38	Examples:
39	dj.kill('HOST LIKE "%compute%"') lists only connections from hosts containing "compute".
40	dj.kill('TIME > 600') lists only connections in their current state for more than 10 minutes
41	"""
42
43	if connection is None:
44	connection = conn()
45
46	if order_by is not None and not isinstance(order_by, str):
47	order_by = ",".join(order_by)
48
49	query = (
50	"SELECT * FROM information_schema.processlist WHERE id <> CONNECTION_ID()"
51	+ ("" if restriction is None else " AND (%s)" % restriction)
52	+ (" ORDER BY %s" % (order_by or "id"))
53	)
54
55	while True:
56	print(" ID USER HOST STATE TIME INFO")
57	print("+--+ +----------+ +-----------+ +-----------+ +-----+")
58	cur = (
59	{k.lower(): v for k, v in elem.items()}
60	for elem in connection.query(query, as_dict=True)
61	)
62	for process in cur:
63	try:
64	print(
65	"{id:>4d} {user:<12s} {host:<12s} {state:<12s} {time:>7d} {info}".format(
66	**process
67	)
68	)
69	except TypeError:
70	print(process)
71	response = input('process to kill or "q" to quit > ')
72	if response == "q":
73	break
74	if response:
75	try:
76	pid = int(response)
77	except ValueError:
78	pass # ignore non-numeric input
79	else:
80	try:
81	connection.query("kill %d" % pid)
82	except pymysql.err.InternalError:
83	print("Process not found")
84
85
86	def kill_quick(restriction=None, connection=None):	1✔
87	"""
88	Kill database connections without prompting. Returns number of terminated connections.
89	:param restriction: restriction to be applied to processlist
90	:param connection: a datajoint.Connection object. Default calls datajoint.conn()
91
92	Restrictions are specified as strings and can involve any of the attributes of
93	information_schema.processlist: ID, USER, HOST, DB, COMMAND, TIME, STATE, INFO.
94
95	Examples:
96	dj.kill('HOST LIKE "%compute%"') terminates connections from hosts containing "compute".
97	"""
98	if connection is None:	×
99	connection = conn()	×
100
101	query = (	×
102	"SELECT * FROM information_schema.processlist WHERE id <> CONNECTION_ID()"
103	+ ("" if restriction is None else " AND (%s)" % restriction)
104	)
105
106	cur = (	×
107	{k.lower(): v for k, v in elem.items()}
108	for elem in connection.query(query, as_dict=True)
109	)
110	nkill = 0	×
111	for process in cur:	×
112	connection.query("kill %d" % process["id"])	×
113	nkill += 1	×
114	return nkill	×

datajoint / datajoint-python / #11916

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous