From bd5061b4b3482800be1a66b2299c4b776733ee5c Mon Sep 17 00:00:00 2001 From: +Sam Thongo <+samuelthongo73@gmail.com> Date: Sun, 5 Nov 2023 20:24:47 +0300 Subject: [PATCH] freebies --- alembic.ini | 116 ++++++++++++++++++ lib/__pycache__/models.cpython-38.pyc | Bin 0 -> 3944 bytes lib/alembic/README | 1 + lib/alembic/env.py | 78 ++++++++++++ lib/alembic/script.py.mako | 26 ++++ lib/freebies.db | Bin 20480 -> 24576 bytes lib/migrations/__pycache__/env.cpython-38.pyc | Bin 0 -> 1755 bytes .../versions/7a71dbf71c64_create_db.py | 17 ++- ...bf48c_create_companies_devs.cpython-38.pyc | Bin 0 -> 1050 bytes .../7a71dbf71c64_create_db.cpython-38.pyc | Bin 0 -> 1130 bytes lib/models.py | 66 +++++++++- lib/seed.py | 49 +++++++- 12 files changed, 347 insertions(+), 6 deletions(-) create mode 100644 alembic.ini create mode 100644 lib/__pycache__/models.cpython-38.pyc create mode 100644 lib/alembic/README create mode 100644 lib/alembic/env.py create mode 100644 lib/alembic/script.py.mako create mode 100644 lib/migrations/__pycache__/env.cpython-38.pyc create mode 100644 lib/migrations/versions/__pycache__/5f72c58bf48c_create_companies_devs.cpython-38.pyc create mode 100644 lib/migrations/versions/__pycache__/7a71dbf71c64_create_db.cpython-38.pyc diff --git a/alembic.ini b/alembic.ini new file mode 100644 index 000000000..1e8c25827 --- /dev/null +++ b/alembic.ini @@ -0,0 +1,116 @@ +# A generic, single database configuration. + +[alembic] +# path to migration scripts +script_location = alembic + +# template used to generate migration file names; The default value is %%(rev)s_%%(slug)s +# Uncomment the line below if you want the files to be prepended with date and time +# see https://alembic.sqlalchemy.org/en/latest/tutorial.html#editing-the-ini-file +# for all available tokens +# file_template = %%(year)d_%%(month).2d_%%(day).2d_%%(hour).2d%%(minute).2d-%%(rev)s_%%(slug)s + +# sys.path path, will be prepended to sys.path if present. +# defaults to the current working directory. +prepend_sys_path = . + +# timezone to use when rendering the date within the migration file +# as well as the filename. +# If specified, requires the python-dateutil library that can be +# installed by adding `alembic[tz]` to the pip requirements +# string value is passed to dateutil.tz.gettz() +# leave blank for localtime +# timezone = + +# max length of characters to apply to the +# "slug" field +# truncate_slug_length = 40 + +# set to 'true' to run the environment during +# the 'revision' command, regardless of autogenerate +# revision_environment = false + +# set to 'true' to allow .pyc and .pyo files without +# a source .py file to be detected as revisions in the +# versions/ directory +# sourceless = false + +# version location specification; This defaults +# to alembic/versions. When using multiple version +# directories, initial revisions must be specified with --version-path. +# The path separator used here should be the separator specified by "version_path_separator" below. +# version_locations = %(here)s/bar:%(here)s/bat:alembic/versions + +# version path separator; As mentioned above, this is the character used to split +# version_locations. The default within new alembic.ini files is "os", which uses os.pathsep. +# If this key is omitted entirely, it falls back to the legacy behavior of splitting on spaces and/or commas. +# Valid values for version_path_separator are: +# +# version_path_separator = : +# version_path_separator = ; +# version_path_separator = space +version_path_separator = os # Use os.pathsep. Default configuration used for new projects. + +# set to 'true' to search source files recursively +# in each "version_locations" directory +# new in Alembic version 1.10 +# recursive_version_locations = false + +# the output encoding used when revision files +# are written from script.py.mako +# output_encoding = utf-8 + +sqlalchemy.url = driver://user:pass@localhost/dbname + + +[post_write_hooks] +# post_write_hooks defines scripts or Python functions that are run +# on newly generated revision scripts. See the documentation for further +# detail and examples + +# format using "black" - use the console_scripts runner, against the "black" entrypoint +# hooks = black +# black.type = console_scripts +# black.entrypoint = black +# black.options = -l 79 REVISION_SCRIPT_FILENAME + +# lint with attempts to fix using "ruff" - use the exec runner, execute a binary +# hooks = ruff +# ruff.type = exec +# ruff.executable = %(here)s/.venv/bin/ruff +# ruff.options = --fix REVISION_SCRIPT_FILENAME + +# Logging configuration +[loggers] +keys = root,sqlalchemy,alembic + +[handlers] +keys = console + +[formatters] +keys = generic + +[logger_root] +level = WARN +handlers = console +qualname = + +[logger_sqlalchemy] +level = WARN +handlers = +qualname = sqlalchemy.engine + +[logger_alembic] +level = INFO +handlers = +qualname = alembic + +[handler_console] +class = StreamHandler +args = (sys.stderr,) +level = NOTSET +formatter = generic + +[formatter_generic] +format = %(levelname)-5.5s [%(name)s] %(message)s +datefmt = %H:%M:%S diff --git a/lib/__pycache__/models.cpython-38.pyc b/lib/__pycache__/models.cpython-38.pyc new file mode 100644 index 0000000000000000000000000000000000000000..c6b460ba8c8cb3c580cb8984030e46c3d0c171ec GIT binary patch literal 3944 zcmcgvPmkNi73T~|iK70jcC&VDH*J)p2_mFkxph&r$<{Dx!+}v`(PG^TfrQ{{M!Vc4 zMR~~GK!SW)AL~QUK>+*c?@*xMW3D~*E9BDt-jLF2V;R_7N@1Sf4ByPWzrPv&t=;x5 zJU9RKrTlxtvi?n-)yF~SK3e`y5Mc?HSSbrw$^)L-ft@;mle&S+=p9dLX+5Z?Uf`vT zpkc;rjQN3Ycn5QvK@+?yYB391(v>Yyf5C#b^krLk;5uJgq9Od}mhffg+zvL-Z;BTB zEz|F!-xeM8JEp%Wn_}af1zWOn+TmrA7Fc~( z`;RgusEGQuZYh~WMVyWEqj;k0 z2hs3E$&qg5GS4xZMkiABneGZXOdgP>v7dvDAEYGW%k55`O$? z5gjBl97n0_=jaZN8sVLb5$qt9k|JD{#EP4k0|mojHa?SM(h7rK3T*|n9g>Z!&oX}B zU{qQG6IQ?l3vB5~_Z%@3_6r`=(5s(Y@S|6bgKG#!xX-P?7d27Ge^Yp(f&UgF<%{NX zlJ1C>Xk%L{5inMLpuFJwVHAAam>3 z+P7Y_JXM*Qw+luv z!JiwYAHp4QfxUNW0twq_`)K(V$RUKk_MfpiJGTD8pE~=cNeH1W!rO|oH>C^?m=s`{fcyxNY0&^4Hq#znuTNEuzPYRk-2Jq)Q2i2R($k4$JCQo&U|qn%79 zS**i+Hrrm9vnq5cMwEXPBV!i1sIXqLg1=%f?SdaW_;+8y5cLjTLA{Uhr`@NgQccwc z@m{8cRN=u?w@2COSWt0JWu&w_id9~$^?D~ugv<*Ru_>4Lu}R|D5uiw>>q*cOD`(cP0 zIt7A-x)p{`Pot!I(+*3(D!Ymtt0+%3kA<1?c^g@JJD=QrH&;w=b}lbolTI~tVO$C;73PMmZ)D7 zAs4EC(ihG-JLf@j*>_E!E=cCeL>DBw*3z{J{6>Wu-3m1}OVm&wLIIPHC2Gv- z2yMO-i&^WUhGKDLFu`5IG1;*$Mt{IwkI8BiBR*?Yhwu)~_$ga>i?eHODtwMDzFOI) zyWFNik~g-hL)sKFg^>#RnBh31V~x|XhJV|P*x;P+tncdpTxW0Q^|Eirc`xeCYQ2%l z(%x*d_ibDpVZ_`sdfBMgcNLjk9y~qbNWJntvBpQbjoa$DfJ2HXPV&D(c9GH77G_+M zc*7POQ;aR^i>a|&EmIa$pOx6NMirCrPgWGT@MCpm@AbE?Tvk0SR9a2J4{C^b7>mHK zLIPfU6O~P*DyF5!jGw-sX>UD=DS7Xs<#aeNPh!?(T~6mPHP1FFKD(AbO>hPLd@vY{ zs+;|eIG}Vrh{s~De_i{Kh?3!vOs9%;TtNo|40Hq4oeQCzOr?r4M-dz=e7{0|`beU8_FL{OU_BN)lb)FTd=}WpCxYTV6o4w_DzS?$ME+ f(-HU{+h#4+vbn|o>)1Z`dG}`bR`*>Gqdxl|NN*v> literal 0 HcmV?d00001 diff --git a/lib/alembic/README b/lib/alembic/README new file mode 100644 index 000000000..98e4f9c44 --- /dev/null +++ b/lib/alembic/README @@ -0,0 +1 @@ +Generic single-database configuration. \ No newline at end of file diff --git a/lib/alembic/env.py b/lib/alembic/env.py new file mode 100644 index 000000000..36112a3c6 --- /dev/null +++ b/lib/alembic/env.py @@ -0,0 +1,78 @@ +from logging.config import fileConfig + +from sqlalchemy import engine_from_config +from sqlalchemy import pool + +from alembic import context + +# this is the Alembic Config object, which provides +# access to the values within the .ini file in use. +config = context.config + +# Interpret the config file for Python logging. +# This line sets up loggers basically. +if config.config_file_name is not None: + fileConfig(config.config_file_name) + +# add your model's MetaData object here +# for 'autogenerate' support +# from myapp import mymodel +# target_metadata = mymodel.Base.metadata +target_metadata = None + +# other values from the config, defined by the needs of env.py, +# can be acquired: +# my_important_option = config.get_main_option("my_important_option") +# ... etc. + + +def run_migrations_offline() -> None: + """Run migrations in 'offline' mode. + + This configures the context with just a URL + and not an Engine, though an Engine is acceptable + here as well. By skipping the Engine creation + we don't even need a DBAPI to be available. + + Calls to context.execute() here emit the given string to the + script output. + + """ + url = config.get_main_option("sqlalchemy.url") + context.configure( + url=url, + target_metadata=target_metadata, + literal_binds=True, + dialect_opts={"paramstyle": "named"}, + ) + + with context.begin_transaction(): + context.run_migrations() + + +def run_migrations_online() -> None: + """Run migrations in 'online' mode. + + In this scenario we need to create an Engine + and associate a connection with the context. + + """ + connectable = engine_from_config( + config.get_section(config.config_ini_section, {}), + prefix="sqlalchemy.", + poolclass=pool.NullPool, + ) + + with connectable.connect() as connection: + context.configure( + connection=connection, target_metadata=target_metadata + ) + + with context.begin_transaction(): + context.run_migrations() + + +if context.is_offline_mode(): + run_migrations_offline() +else: + run_migrations_online() diff --git a/lib/alembic/script.py.mako b/lib/alembic/script.py.mako new file mode 100644 index 000000000..fbc4b07dc --- /dev/null +++ b/lib/alembic/script.py.mako @@ -0,0 +1,26 @@ +"""${message} + +Revision ID: ${up_revision} +Revises: ${down_revision | comma,n} +Create Date: ${create_date} + +""" +from typing import Sequence, Union + +from alembic import op +import sqlalchemy as sa +${imports if imports else ""} + +# revision identifiers, used by Alembic. +revision: str = ${repr(up_revision)} +down_revision: Union[str, None] = ${repr(down_revision)} +branch_labels: Union[str, Sequence[str], None] = ${repr(branch_labels)} +depends_on: Union[str, Sequence[str], None] = ${repr(depends_on)} + + +def upgrade() -> None: + ${upgrades if upgrades else "pass"} + + +def downgrade() -> None: + ${downgrades if downgrades else "pass"} diff --git a/lib/freebies.db b/lib/freebies.db index 12beb1c963e832db481e7a7493e3029e691ac4dc..47c2eb87e577842378b0100f53750419374a24dd 100644 GIT binary patch delta 598 zcmZozz}Rqrae}lU9|Hpe8xX?)%S0VxaXtn;`Gi z$D}EOm0es`ma&z&Brz!`HLWN$H7PT-7)G->2e~?ixGID=I{CONKt(4z^2yj{mZat? zggFK|dpHK^C~$F>CFYc-DtP*ZxVpQ71yWMW;xki_MUwM#3lj4xfufV&^U2hMbUFL` z1&0JV0<|fmWyiy8jYp`2GJ#G~aPtpx^>p`B@OF*VKq$~u2y%6E4RZB!b`4g52!j-B zWTt51u>j~eW1u_Yfi$YY5LcrZ3RVL$k4tm26yGgD9%CcZq!eSzq|`)%#7zoc_@$5q zc!5FB%zu`F|2_Xb{4F8MJ*QqJ`Wae}lUD+2=q2*UvLL>*&MRtCLzSzi7h3@m(74E((OJNfGQq&5o*EaTlQ r#dk}P$285{DB08^Db2(pd6U8yej#K5UXW1&K-Ks8Cr*%NL*)VhkJ1_o diff --git a/lib/migrations/__pycache__/env.cpython-38.pyc b/lib/migrations/__pycache__/env.cpython-38.pyc new file mode 100644 index 0000000000000000000000000000000000000000..c45593b4b6c369fc9f0dd400f54db2949d8d00d8 GIT binary patch literal 1755 zcmZuxL5~|X6t*XmNixZ1+d`{~1Nc@VbvF_foDeE)sZ>C#wzL98NF!(bk~r(}47Rh| zt=dyLz@e(dnM3x-pWz40l@osff#BJbbhm&<_B=n^?|skjdw#sLlM-xS{CXD-yM+Aa zC+AxWCpWR%pKuUFP(=zlq14Nfii*~x733Dm@gxp%tlC8~NhpD2*_m{pv+RoYVb6RH z)XOiL(~0U8=_JKn>9Ri=h_>kbOhs4pzK`d zBBR0Yh>MJ9ozA3!+ohh#InFMDo=Xkc%#=lzZ?$n%mI_xpsB!Sb9iSZF;ui3q7;r12 zb8l|jC<;C7x3SxsI5@JVf}Yan#6`>2GvZf_o^W3*P3^TeQvu_odebk2W{5a;#jrYUMUYpAPMJimQA9 z#mWB4s7G~Lal;GiP88H_%?l8txNhN6z3aG{gUbqVTyV$h9VHzYuCl4r!qx*Jxq{qf zW##O_s8c7+z1O?p6qnksV75#fX;AlGXT|jW{NfalZkaP&Tb}!U_0X(zc9!H3HNMb( zA?7Xo{qbPDC<_={UX1VHC91M2K)Z2{$sbn>WaN!;b%M#$d)1A-nE{we*hBOAA(*j} z)A5<*G3cZH>ZHE%ro(K*sCUtwy@Uf9q%@{c_>w=9PD;&X)V@LZ2o)Y?ApEx|IO6C! za&*}`MdG`7*%r~u)^qZVtdZWYK6p+YR`fIKdcF)^;#=T4>vU5Bopt}s5;#{3zIKzR zKmK1O=>HVLJ&i;_=Gq)IH?l;oHmviE^29{OZ8`tC61cTxF2kCqoyP29UYf!=S4mqk z_0N%M6xB&(U?!i`J^y6n3XMG)wd(ZDP?Pvw16qK|xXq^A<%@&Sz_jq*%n*mV@5y5u z$MvO#Gnv$Kt243R)je1#bsyb?cgf=nBr_Mh9ZxTF(GNa{2&}(>;H%4A^$VU*&G0r# z_2qx_Al&CixMbhKfpj98A_?NZk|Yp None: - pass + # ### commands auto generated by Alembic - please adjust! ### + op.create_table('freebies', + sa.Column('id', sa.Integer(), nullable=False), + sa.Column('item', sa.String(), nullable=True), + sa.Column('value', sa.Integer(), nullable=True), + sa.Column('dev_id', sa.Integer(), nullable=True), + sa.Column('company_id', sa.Integer(), nullable=True), + sa.ForeignKeyConstraint(['company_id'], ['companies.id'], name=op.f('fk_freebies_company_id_companies')), + sa.ForeignKeyConstraint(['dev_id'], ['devs.id'], name=op.f('fk_freebies_dev_id_devs')), + sa.PrimaryKeyConstraint('id') + ) + # ### end Alembic commands ### def downgrade() -> None: - pass + # ### commands auto generated by Alembic - please adjust! ### + op.drop_table('freebies') + # ### end Alembic commands ### \ No newline at end of file diff --git a/lib/migrations/versions/__pycache__/5f72c58bf48c_create_companies_devs.cpython-38.pyc b/lib/migrations/versions/__pycache__/5f72c58bf48c_create_companies_devs.cpython-38.pyc new file mode 100644 index 0000000000000000000000000000000000000000..2d592da33ac117bdabae0086ba902aa349d970d0 GIT binary patch literal 1050 zcma)5&2G~`5cXee$4yWv72+6)QzTO7w5R8A+MuG!t?RaN)=JU??tzS2rj)LRt>k0hU73Ev*Y!8en&v1(8 zXo%t|L?v3RCYq9<9x>kl(s50mY#c!ZDg;pjfg2G zhiu7dm{)9%oT@2K@c~VyT&qOQv^P4@GVR*fDA=g+NiQNbjMOk<-bAb_r;1#gMBAvF znp)!6|A$jm8A!c?G}g~${f0oPMdSnzZ_{NQ2eTNYL4yQd$>y9bDT(N%zlMNIBj>qo zaSeNrjZy|bD+)9bNOQ{KWkRKON;9|wF3pRKW3k+S!J{PL>oZtS85J21BAR_s_%7lU z=EvGyumoHYB2nz)cr{fPtH?vjwA_bkLb|fLR0`#0lR5cjIX0CebAi zG8A99E8zJp5bm~6es$WLp8dR+gLEyeWs0Q)?&a#-N26$_x>r2*Ps1$N_AkDNJAcWV z-|<4$2e>F@-Kdo?C{jhXe$MN-&kZzft)$N(oTVjsw>*@qyO<{HguSE1!5*S3@=`@} zRKAbz_KviCpRmyPr4_`G%%ZSD6<2W(hYLv7d7reHvH^rM>F5f|IWTzTXQv9AHE*rv8id_3So5_Bs``gpArFW zqyYmxW{E!1S-~5zF*3m%HNY5IbZj~f7|{64zykZDIcnCD15J`#FwV4&d+|CH806T7 zvFpCTc_gEhw@;3S?StT;2jlaDUbz3Dxwc#^wGLBynw_fFsy6~Hf%#1Eu+Uju8zdakk(Wer?UDh@^*09SxqfLb zHLAe;WRysQkOm>v6=K&q*NF2E()&F96pQVU>u_|qa2f$sxc4Z87d2K?nj{9Fio<6FNnWr zRXkbmL@;Wfzx986^RKmPmXyf5$%6zhZ1Po_Y^|D90sP{w(-b;2P_?SZ_46_zT?WcJ zOygO?m3_iEa&I&JAVzKBS@VD?5lW z8AoC5Q@)FXIGkd#5Jd*c5%rH#YlS=iS{#djhf|*>31g{TzzjJ^Kjmd%Wt{(}' @@ -25,5 +41,51 @@ class Dev(Base): id = Column(Integer(), primary_key=True) name= Column(String()) + def freebies(self): + return self.freebies + + def companies(self): + return [freebie.company for freebie in self.freebies] + + def received_freebie(self,item): + for freebie in self.freebies: + if freebie.item == item: + return True + return False + + def give_freebie(self, session, other_dev, freebie): + if freebie.dev == self: + freebie.dev = other_dev + session.commit() + def __repr__(self): return f'' + +class Freebie(Base): + __tablename__ = 'freebies' + + id = Column(Integer(), primary_key=True) + item = Column(String()) + value = Column(Integer()) + dev_id = Column(Integer(), ForeignKey('devs.id')) + company_id = Column(Integer(), ForeignKey('companies.id')) + + @property + def _dev(self): + return self._dev + + @property + def _company(self): + return self._company + + def print_details(self): + return f"{self._dev.name} owns a {self.item} from {self._company.name} with a value of {self.value}" + + def __repr__(self): + return f'' + +engine = create_engine("sqlite:///freebies.db") +Base.metadata.create_all(engine) +Session = sessionmaker(bind=engine) +session = Session() + diff --git a/lib/seed.py b/lib/seed.py index b16becbbb..eb799cda0 100644 --- a/lib/seed.py +++ b/lib/seed.py @@ -1,3 +1,48 @@ -#!/usr/bin/env python3 +from sqlalchemy import create_engine +from sqlalchemy.orm import sessionmaker +from models import Base, Company, Dev, Freebie -# Script goes here! +engine = create_engine("sqlite:///freebies.db") +Base.metadata.create_all(engine) +Session = sessionmaker(bind=engine) +session = Session() + +# Clear the existing data +session.query(Company).delete() +session.query(Dev).delete() +session.query(Freebie).delete() +session.commit() + +# Sample data +company1 = Company(name="Company A", founding_year=1990) +company2 = Company(name="Company B", founding_year=2000) +company3 = Company(name="Company C", founding_year=2010) + +dev1 = Dev(name="Dev 1") +dev2 = Dev(name="Dev 2") +dev3 = Dev(name="Dev 3") + +# Create Freebie instances and associate them with Dev and Company +freebie1 = Freebie(item="T-shirt", value=10) +freebie1.dev = dev1 +freebie1.company = company1 + +freebie2 = Freebie(item="Sticker", value=2) +freebie2.dev = dev2 +freebie2.company = company2 + +freebie3 = Freebie(item="Mug", value=5) +freebie3.dev = dev1 +freebie3.company = company3 + +freebie4 = Freebie(item="T-shirt", value=15) +freebie4.dev = dev3 +freebie4.company = company2 + +# Add and commit the sample data +session.add_all([company1, company2, company3, dev1, dev2, dev3, freebie1, freebie2, freebie3, freebie4]) +session.commit() + +# Print the Freebie instances +for freebie in session.query(Freebie).all(): + print(freebie)