#12888

pending completion

Build # #12888

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/97821871dd4afcc3b1af0b6e38320710db4a4dd7">97821871d<a href="https://github.com/datajoint/datajoint-python/commit/97821871dd4afcc3b1af0b6e38320710db4a4dd7">">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/32504b17513c64aa1895146c80609d76af22bd1c">32504b175</a>">32504b175</a><a href="https://github.com/datajoint/datajoint-python/commit/97821871dd4afcc3b1af0b6e38320710db4a4dd7"> into 3b6e84588">3b6e84588</a>

Run Details

8 of 8 new or added lines in 1 file covered. (100.0%)

3108 of 3432 relevant lines covered (90.56%)

0.91 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
32	:param restriction: restriction to be applied to processlist
33	:param connection: a datajoint.Connection object. Default calls datajoint.conn()
34	:param order_by: order by a single attribute or the list of attributes. defaults to 'id'.
35
36	Restrictions are specified as strings and can involve any of the attributes of
37	information_schema.processlist: ID, USER, HOST, DB, COMMAND, TIME, STATE, INFO.
38
39	Examples:
40	dj.kill('HOST LIKE "%compute%"') lists only connections from hosts containing "compute".
41	dj.kill('TIME > 600') lists only connections in their current state for more than 10 minutes
42	"""
43
44	if connection is None:
45	connection = conn()
46
47	if order_by is not None and not isinstance(order_by, str):
48	order_by = ",".join(order_by)
49
50	query = (
51	"SELECT * FROM information_schema.processlist WHERE id <> CONNECTION_ID()"
52	+ ("" if restriction is None else " AND (%s)" % restriction)
53	+ (" ORDER BY %s" % (order_by or "id"))
54	)
55
56	while True:
57	print(" ID USER HOST STATE TIME INFO")
58	print("+--+ +----------+ +-----------+ +-----------+ +-----+")
59	cur = (
60	{k.lower(): v for k, v in elem.items()}
61	for elem in connection.query(query, as_dict=True)
62	)
63	for process in cur:
64	try:
65	print(
66	"{id:>4d} {user:<12s} {host:<12s} {state:<12s} {time:>7d} {info}".format(
67	**process
68	)
69	)
70	except TypeError:
71	print(process)
72	response = input('process to kill or "q" to quit > ')
73	if response == "q":
74	break
75	if response:
76	try:
77	pid = int(response)
78	except ValueError:
79	pass # ignore non-numeric input
80	else:
81	try:
82	connection.query("kill %d" % pid)
83	except pymysql.err.InternalError:
84	print("Process not found")
85
86
87	def kill_quick(restriction=None, connection=None):	1✔
88	"""
89	Kill database connections without prompting. Returns number of terminated connections.
90
91	:param restriction: restriction to be applied to processlist
92	:param connection: a datajoint.Connection object. Default calls datajoint.conn()
93
94	Restrictions are specified as strings and can involve any of the attributes of
95	information_schema.processlist: ID, USER, HOST, DB, COMMAND, TIME, STATE, INFO.
96
97	Examples:
98	dj.kill('HOST LIKE "%compute%"') terminates connections from hosts containing "compute".
99	"""
100	if connection is None:	×
101	connection = conn()	×
102
103	query = (	×
104	"SELECT * FROM information_schema.processlist WHERE id <> CONNECTION_ID()"
105	+ ("" if restriction is None else " AND (%s)" % restriction)
106	)
107
108	cur = (	×
109	{k.lower(): v for k, v in elem.items()}
110	for elem in connection.query(query, as_dict=True)
111	)
112	nkill = 0	×
113	for process in cur:	×
114	connection.query("kill %d" % process["id"])	×
115	nkill += 1	×
116	return nkill	×

datajoint / datajoint-python / #12888

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