| 1 | | {{{#!diff |
| 2 | | diff --git a/docs/ref/schema-editor.txt b/docs/ref/schema-editor.txt |
| 3 | | index 54dd3bf..c503296 100644 |
| 4 | | --- a/docs/ref/schema-editor.txt |
| 5 | | +++ b/docs/ref/schema-editor.txt |
| 6 | | @@ -158,9 +158,9 @@ below. |
| 7 | | |
| 8 | | .. attribute:: SchemaEditor.connection |
| 9 | | |
| 10 | | -A connection object to the database. A useful attribute of the |
| 11 | | -connection is ``alias`` which can be used to determine the name of |
| 12 | | -the database being accessed. |
| 13 | | +A connection object to the database. A useful attribute of the connection is |
| 14 | | +``alias`` which can be used to determine the name of the database being |
| 15 | | +accessed. |
| 16 | | |
| 17 | | -This in turn is useful when doing data migrations for |
| 18 | | -:ref:`migrations with multiple database backends <data-migrations-and-multiple-databases>`. |
| 19 | | +This in turn is useful when doing data migrations for :ref:`migrations with |
| 20 | | +multiple database backends <data-migrations-and-multiple-databases>`. |
| 21 | | diff --git a/docs/topics/migrations.txt b/docs/topics/migrations.txt |
| 22 | | index 746f4d2..a743bf2 100644 |
| 23 | | --- a/docs/topics/migrations.txt |
| 24 | | +++ b/docs/topics/migrations.txt |
| 25 | | @@ -472,71 +472,72 @@ backwards will raise an exception. |
| 26 | | Data migrations and multiple databases |
| 27 | | ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ |
| 28 | | |
| 29 | | -One of the challenges that can be encountered is figuring out whether |
| 30 | | -or not to run a migration when using multiple databases. |
| 31 | | - |
| 32 | | -For example, you may want to **only** run a migration against a |
| 33 | | -particular database. |
| 34 | | - |
| 35 | | -In order to do that you can filter inside of your data migration |
| 36 | | -by looking at the ``schema_editor.connection.alias`` attribute. |
| 37 | | -However it is better to have your ``dbrouter`` leverage the same |
| 38 | | -``allow_migrate`` call by making the following method in your |
| 39 | | -router:: |
| 40 | | - |
| 41 | | - class MyRouter(object): |
| 42 | | - |
| 43 | | - def allow_migrate(self, db, model): |
| 44 | | - # your typical migration code here |
| 45 | | - |
| 46 | | - def runpython_ok(self, apps, schema_editor, model): |
| 47 | | - db = schema_editor.connection.alias |
| 48 | | - if self.allow_migrate(db, model): |
| 49 | | - return True |
| 50 | | - return False |
| 51 | | - |
| 52 | | -Then to leverage this in your migrations, do the following:: |
| 53 | | - |
| 54 | | - # -*- coding: utf-8 -*- |
| 55 | | - from __future__ import unicode_literals |
| 56 | | - |
| 57 | | - from django.db import models, migrations |
| 58 | | - from myappname.dbrouter import MyRouter |
| 59 | | - import sys |
| 60 | | - |
| 61 | | - def forwards(apps, schema_editor): |
| 62 | | - # only run this for the correct databases |
| 63 | | - MyModel = apps.get_model("myappname", "MyModel") |
| 64 | | - if not MyRouter().runpython_ok(apps, schema_editor, MyModel): |
| 65 | | - return |
| 66 | | - |
| 67 | | - """ |
| 68 | | - This is a global configuration model so there should be only |
| 69 | | - row. |
| 70 | | - Get the first one -or- create the initial one. |
| 71 | | - Only set the default ports if it has not been set by user |
| 72 | | - already. |
| 73 | | - """ |
| 74 | | - try: |
| 75 | | - my_model = MyModel.objects.order_by("-id").first() |
| 76 | | - except IndexError: |
| 77 | | - my_model = MyModel.objects.create() |
| 78 | | - if my_model.my_model_ports == "": |
| 79 | | - my_model.my_model_ports = "80, 8080, 3128" |
| 80 | | - my_model.save() |
| 81 | | - |
| 82 | | - def backwards(apps, schema_editor): |
| 83 | | - pass |
| 84 | | - |
| 85 | | - class Migration(migrations.Migration): |
| 86 | | - |
| 87 | | - dependencies = [ |
| 88 | | - # Dependencies to other migrations |
| 89 | | - ] |
| 90 | | - |
| 91 | | - operations = [ |
| 92 | | - migrations.RunPython(forwards, backwards), |
| 93 | | - ] |
| 94 | | +One of the challenges that can be encountered is figuring out whether or not to |
| 95 | | +run a migration when using multiple databases. |
| 96 | | + |
| 97 | | +For example, you may want to **only** run a migration against a particular |
| 98 | | +database. |
| 99 | | + |
| 100 | | +In order to do that you can check the connection alias inside a ``RunPython`` |
| 101 | | +forwards and backwards operation by looking at the |
| 102 | | +``schema_editor.connection.alias`` attribute. |
| 103 | | + |
| 104 | | + # -*- coding: utf-8 -*- |
| 105 | | + from __future__ import unicode_literals |
| 106 | | + |
| 107 | | + from django.db import migrations |
| 108 | | + |
| 109 | | + def forwards(apps, schema_editor): |
| 110 | | + if not schema_editor.connection.alias == 'master-db': |
| 111 | | + return |
| 112 | | + # Your migration code goes here |
| 113 | | + |
| 114 | | + class Migration(migrations.Migration): |
| 115 | | + |
| 116 | | + dependencies = [ |
| 117 | | + # Dependencies to other migrations |
| 118 | | + ] |
| 119 | | + |
| 120 | | + operations = [ |
| 121 | | + migrations.RunPython(forwards), |
| 122 | | + ] |
| 123 | | + |
| 124 | | +You can also use your database router's ``allow_migrate`` method, but keep in |
| 125 | | +mind that the imported router needs to stay around as long as it is referenced |
| 126 | | +inside a migration: |
| 127 | | + |
| 128 | | +.. snippet:: |
| 129 | | + :filename: myapp/dbrouters.py |
| 130 | | + |
| 131 | | + class MyRouter(object): |
| 132 | | + |
| 133 | | + def allow_migrate(self, db, model): |
| 134 | | + return db == 'default': |
| 135 | | + |
| 136 | | +Then, to leverage this in your migrations, do the following:: |
| 137 | | + |
| 138 | | + # -*- coding: utf-8 -*- |
| 139 | | + from __future__ import unicode_literals |
| 140 | | + |
| 141 | | + from django.db import models, migrations |
| 142 | | + |
| 143 | | + from myappname.dbrouters import MyRouter |
| 144 | | + |
| 145 | | + def forwards(apps, schema_editor): |
| 146 | | + MyModel = apps.get_model("myappname", "MyModel") |
| 147 | | + if not MyRouter().allow_migrate(schema_editor.connection.alias, MyModel): |
| 148 | | + return |
| 149 | | + # Your migration code goes here |
| 150 | | + |
| 151 | | + class Migration(migrations.Migration): |
| 152 | | + |
| 153 | | + dependencies = [ |
| 154 | | + # Dependencies to other migrations |
| 155 | | + ] |
| 156 | | + |
| 157 | | + operations = [ |
| 158 | | + migrations.RunPython(forwards), |
| 159 | | + ] |
| 160 | | |
| 161 | | More advanced migrations |
| 162 | | ~~~~~~~~~~~~~~~~~~~~~~~~ |
| 163 | | }}} |
| | 1 | Some suggested changes. |