commit 58d58edefb875763ad13b4b0122b07dadae36e65 Author: Johnny Gear Date: Thu Nov 13 22:03:20 2025 -0600 Initial commit diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..590f5de --- /dev/null +++ b/.gitignore @@ -0,0 +1,2 @@ +*~ +db.sqlite3 diff --git a/Pipfile b/Pipfile new file mode 100644 index 0000000..ab425e5 --- /dev/null +++ b/Pipfile @@ -0,0 +1,12 @@ +[[source]] +url = "https://pypi.org/simple" +verify_ssl = true +name = "pypi" + +[packages] +pyyaml = "*" + +[dev-packages] + +[requires] +python_version = "3.12" diff --git a/Pipfile.lock b/Pipfile.lock new file mode 100644 index 0000000..f710b97 --- /dev/null +++ b/Pipfile.lock @@ -0,0 +1,101 @@ +{ + "_meta": { + "hash": { + "sha256": "27a55b76dc774eff1af052a9de8b0c6b059a44578b80abeb18dbe190986011e0" + }, + "pipfile-spec": 6, + "requires": { + "python_version": "3.12" + }, + "sources": [ + { + "name": "pypi", + "url": "https://pypi.org/simple", + "verify_ssl": true + } + ] + }, + "default": { + "pyyaml": { + "hashes": [ + "sha256:00c4bdeba853cc34e7dd471f16b4114f4162dc03e6b7afcc2128711f0eca823c", + "sha256:0150219816b6a1fa26fb4699fb7daa9caf09eb1999f3b70fb6e786805e80375a", + "sha256:02893d100e99e03eda1c8fd5c441d8c60103fd175728e23e431db1b589cf5ab3", + "sha256:02ea2dfa234451bbb8772601d7b8e426c2bfa197136796224e50e35a78777956", + "sha256:0f29edc409a6392443abf94b9cf89ce99889a1dd5376d94316ae5145dfedd5d6", + "sha256:10892704fc220243f5305762e276552a0395f7beb4dbf9b14ec8fd43b57f126c", + "sha256:16249ee61e95f858e83976573de0f5b2893b3677ba71c9dd36b9cf8be9ac6d65", + "sha256:1d37d57ad971609cf3c53ba6a7e365e40660e3be0e5175fa9f2365a379d6095a", + "sha256:1ebe39cb5fc479422b83de611d14e2c0d3bb2a18bbcb01f229ab3cfbd8fee7a0", + "sha256:214ed4befebe12df36bcc8bc2b64b396ca31be9304b8f59e25c11cf94a4c033b", + "sha256:2283a07e2c21a2aa78d9c4442724ec1eb15f5e42a723b99cb3d822d48f5f7ad1", + "sha256:22ba7cfcad58ef3ecddc7ed1db3409af68d023b7f940da23c6c2a1890976eda6", + "sha256:27c0abcb4a5dac13684a37f76e701e054692a9b2d3064b70f5e4eb54810553d7", + "sha256:28c8d926f98f432f88adc23edf2e6d4921ac26fb084b028c733d01868d19007e", + "sha256:2e71d11abed7344e42a8849600193d15b6def118602c4c176f748e4583246007", + "sha256:34d5fcd24b8445fadc33f9cf348c1047101756fd760b4dacb5c3e99755703310", + "sha256:37503bfbfc9d2c40b344d06b2199cf0e96e97957ab1c1b546fd4f87e53e5d3e4", + "sha256:3c5677e12444c15717b902a5798264fa7909e41153cdf9ef7ad571b704a63dd9", + "sha256:3ff07ec89bae51176c0549bc4c63aa6202991da2d9a6129d7aef7f1407d3f295", + "sha256:41715c910c881bc081f1e8872880d3c650acf13dfa8214bad49ed4cede7c34ea", + "sha256:418cf3f2111bc80e0933b2cd8cd04f286338bb88bdc7bc8e6dd775ebde60b5e0", + "sha256:44edc647873928551a01e7a563d7452ccdebee747728c1080d881d68af7b997e", + "sha256:4a2e8cebe2ff6ab7d1050ecd59c25d4c8bd7e6f400f5f82b96557ac0abafd0ac", + "sha256:4ad1906908f2f5ae4e5a8ddfce73c320c2a1429ec52eafd27138b7f1cbe341c9", + "sha256:501a031947e3a9025ed4405a168e6ef5ae3126c59f90ce0cd6f2bfc477be31b7", + "sha256:5190d403f121660ce8d1d2c1bb2ef1bd05b5f68533fc5c2ea899bd15f4399b35", + "sha256:5498cd1645aa724a7c71c8f378eb29ebe23da2fc0d7a08071d89469bf1d2defb", + "sha256:5cf4e27da7e3fbed4d6c3d8e797387aaad68102272f8f9752883bc32d61cb87b", + "sha256:5e0b74767e5f8c593e8c9b5912019159ed0533c70051e9cce3e8b6aa699fcd69", + "sha256:5ed875a24292240029e4483f9d4a4b8a1ae08843b9c54f43fcc11e404532a8a5", + "sha256:5fcd34e47f6e0b794d17de1b4ff496c00986e1c83f7ab2fb8fcfe9616ff7477b", + "sha256:5fdec68f91a0c6739b380c83b951e2c72ac0197ace422360e6d5a959d8d97b2c", + "sha256:6344df0d5755a2c9a276d4473ae6b90647e216ab4757f8426893b5dd2ac3f369", + "sha256:64386e5e707d03a7e172c0701abfb7e10f0fb753ee1d773128192742712a98fd", + "sha256:652cb6edd41e718550aad172851962662ff2681490a8a711af6a4d288dd96824", + "sha256:66291b10affd76d76f54fad28e22e51719ef9ba22b29e1d7d03d6777a9174198", + "sha256:66e1674c3ef6f541c35191caae2d429b967b99e02040f5ba928632d9a7f0f065", + "sha256:6adc77889b628398debc7b65c073bcb99c4a0237b248cacaf3fe8a557563ef6c", + "sha256:79005a0d97d5ddabfeeea4cf676af11e647e41d81c9a7722a193022accdb6b7c", + "sha256:7c6610def4f163542a622a73fb39f534f8c101d690126992300bf3207eab9764", + "sha256:7f047e29dcae44602496db43be01ad42fc6f1cc0d8cd6c83d342306c32270196", + "sha256:8098f252adfa6c80ab48096053f512f2321f0b998f98150cea9bd23d83e1467b", + "sha256:850774a7879607d3a6f50d36d04f00ee69e7fc816450e5f7e58d7f17f1ae5c00", + "sha256:8d1fab6bb153a416f9aeb4b8763bc0f22a5586065f86f7664fc23339fc1c1fac", + "sha256:8da9669d359f02c0b91ccc01cac4a67f16afec0dac22c2ad09f46bee0697eba8", + "sha256:8dc52c23056b9ddd46818a57b78404882310fb473d63f17b07d5c40421e47f8e", + "sha256:9149cad251584d5fb4981be1ecde53a1ca46c891a79788c0df828d2f166bda28", + "sha256:93dda82c9c22deb0a405ea4dc5f2d0cda384168e466364dec6255b293923b2f3", + "sha256:96b533f0e99f6579b3d4d4995707cf36df9100d67e0c8303a0c55b27b5f99bc5", + "sha256:9c57bb8c96f6d1808c030b1687b9b5fb476abaa47f0db9c0101f5e9f394e97f4", + "sha256:9c7708761fccb9397fe64bbc0395abcae8c4bf7b0eac081e12b809bf47700d0b", + "sha256:9f3bfb4965eb874431221a3ff3fdcddc7e74e3b07799e0e84ca4a0f867d449bf", + "sha256:a33284e20b78bd4a18c8c2282d549d10bc8408a2a7ff57653c0cf0b9be0afce5", + "sha256:a80cb027f6b349846a3bf6d73b5e95e782175e52f22108cfa17876aaeff93702", + "sha256:b30236e45cf30d2b8e7b3e85881719e98507abed1011bf463a8fa23e9c3e98a8", + "sha256:b3bc83488de33889877a0f2543ade9f70c67d66d9ebb4ac959502e12de895788", + "sha256:b865addae83924361678b652338317d1bd7e79b1f4596f96b96c77a5a34b34da", + "sha256:b8bb0864c5a28024fac8a632c443c87c5aa6f215c0b126c449ae1a150412f31d", + "sha256:ba1cc08a7ccde2d2ec775841541641e4548226580ab850948cbfda66a1befcdc", + "sha256:bdb2c67c6c1390b63c6ff89f210c8fd09d9a1217a465701eac7316313c915e4c", + "sha256:c1ff362665ae507275af2853520967820d9124984e0f7466736aea23d8611fba", + "sha256:c2514fceb77bc5e7a2f7adfaa1feb2fb311607c9cb518dbc378688ec73d8292f", + "sha256:c3355370a2c156cffb25e876646f149d5d68f5e0a3ce86a5084dd0b64a994917", + "sha256:c458b6d084f9b935061bc36216e8a69a7e293a2f1e68bf956dcd9e6cbcd143f5", + "sha256:d0eae10f8159e8fdad514efdc92d74fd8d682c933a6dd088030f3834bc8e6b26", + "sha256:d76623373421df22fb4cf8817020cbb7ef15c725b9d5e45f17e189bfc384190f", + "sha256:ebc55a14a21cb14062aa4162f906cd962b28e2e9ea38f9b4391244cd8de4ae0b", + "sha256:eda16858a3cab07b80edaf74336ece1f986ba330fdb8ee0d6c0d68fe82bc96be", + "sha256:ee2922902c45ae8ccada2c5b501ab86c36525b883eff4255313a253a3160861c", + "sha256:efd7b85f94a6f21e4932043973a7ba2613b059c4a000551892ac9f1d11f5baf3", + "sha256:f7057c9a337546edc7973c0d3ba84ddcdf0daa14533c2065749c9075001090e6", + "sha256:fa160448684b4e94d80416c0fa4aac48967a969efe22931448d853ada8baf926", + "sha256:fc09d0aa354569bc501d4e787133afc08552722d3ab34836a80547331bb5d4a0" + ], + "index": "pypi", + "markers": "python_version >= '3.8'", + "version": "==6.0.3" + } + }, + "develop": {} +} diff --git a/db.py b/db.py new file mode 100644 index 0000000..a5dc9cf --- /dev/null +++ b/db.py @@ -0,0 +1,48 @@ +import os +import sqlite3 + +SQLITE_DB = os.environ.get('SQLITE_DB', 'db.sqlite3') + +TABLE_REPEATS = ''' + CREATE TABLE IF NOT EXISTS repeats ( + id INTEGER PRIMARY KEY AUTOINCREMENT, + probability FLOAT NOT NULL, + orders TEXT NOT NULL, + count INTEGER DEFAULT 0 + ); +''' + +class Database: + def __init__(self): + self.conn = sqlite3.connect(SQLITE_DB, detect_types=sqlite3.PARSE_DECLTYPES) + self.conn.row_factory = sqlite3.Row + self.table_init(TABLE_REPEATS) + + def table_init(self, table_sql): + c = self.conn.cursor() + c.execute(table_sql) + self.conn.commit() + + def repeat_get(self): + c = self.conn.cursor() + sql = 'SELECT id, probability, orders, count FROM repeats LIMIT 1' + c.execute(sql) + return c.fetchone() + + def repeat_increment(self): + c = self.conn.cursor() + sql = 'UPDATE repeats SET count = count + 1' + c.execute(sql) + self.conn.commit() + + def repeat_put(self, probability, orders): + c = self.conn.cursor() + sql = 'INSERT INTO repeats (probability, orders) VALUES (?, ?);' + c.execute(sql, [probability, orders]) + self.conn.commit() + + def repeat_clear(self): + c = self.conn.cursor() + sql = 'DELETE FROM repeats' + c.execute(sql) + self.conn.commit() diff --git a/generate.py b/generate.py new file mode 100644 index 0000000..12b1f08 --- /dev/null +++ b/generate.py @@ -0,0 +1,61 @@ +import yaml, json +import random +from db import Database + +def pick_order(orders): + picked = random.choices(orders, + weights=[o['weight'] for o in orders], + k=1)[0] + + result = [] + repeat = 0.0 + + if('text' in picked): + result.append(picked['text']) + + if('repeat' in picked): + repeat = picked['repeat'] + + if('add' in picked): + for addition in picked['add']: + if addition['probability'] > random.random(): + result.append(addition['text']) + + if('pick' in picked): + (new_result, new_repeat) = pick_order(picked['pick']) + result += new_result + if new_repeat > 0.0: + repeat = new_repeat + + return (result, repeat) + +def generate(filename="orders.yml"): + with open(filename) as stream: + # Do we want to repeat? + db = Database() + repeat = db.repeat_get() + if repeat is not None: + if repeat['probability'] > random.random(): + db.repeat_increment() + return { + "orders": json.loads(repeat['orders']), + "count": repeat['count'] + } + else: + db.repeat_clear() + + # Pick a new order + try: + orders = yaml.safe_load(stream) + except yaml.YAMLError as exc: + print(exc) + + (result, repeat_p) = pick_order(orders['orders']) + + # Log the repeat + if repeat_p > 0.0: + db.repeat_put(repeat_p, json.dumps(result)) + + return { + "orders": result + } diff --git a/main.py b/main.py new file mode 100644 index 0000000..15cbd14 --- /dev/null +++ b/main.py @@ -0,0 +1,19 @@ +import argparse + +from generate import generate + +if __name__=='__main__': + parser = argparse.ArgumentParser(description="Generate gear orders", prog='PROG') + + subparsers = parser.add_subparsers(help="Sub-command help", dest="command") + + parser_immediate = subparsers.add_parser('immediate', help='Immediately generate a command') + + args = parser.parse_args() + + if args.command == 'immediate': + orders_info = generate() + + if "count" in orders_info and orders_info['count'] > 1: + print(f"These are the same orders from the last {orders_info['count']} days") + print("\n".join(orders_info['orders'])) diff --git a/orders.yml b/orders.yml new file mode 100644 index 0000000..0adfa6a --- /dev/null +++ b/orders.yml @@ -0,0 +1,34 @@ +orders: + - weight: 30 + text: "No orders today" + - weight: 70 + text: "Here are your orders:" + pick: + - weight: 20 + text: "MX Pants + MX Boots" + repeat: 0.7 + add: + - probability: 0.5 + text: "+ Collar" + - weight: 20 + text: "Race Pants + MX Boots" + repeat: 0.7 + add: + - probability: 0.5 + text: "+ Collar" + - weight: 40 + text: "Marine Uniform" + repeat: 0.7 + add: + - probability: 0.9 + text: "+ Leg shackles" + - probability: 0.5 + text: "+ Collar" + - weight: 30 + text: "Army Uniform" + repeat: 0.7 + add: + - probability: 0.9 + text: "+ Leg Shackles" + - probability: 0.5 + text: "+ Collar"